Spaces:
Configuration error
Configuration error
| { | |
| "name": "cifar10_ELR_plus_PreActResNet18", | |
| "n_gpu": 1, | |
| "seed":123, | |
| "arch": { | |
| "args": {"num_classes":10} | |
| }, | |
| "arch1": { | |
| "type": "PreActResNet18", | |
| "args": {"num_classes":10} | |
| }, | |
| "arch2": { | |
| "type": "PreActResNet18", | |
| "args": {"num_classes":10} | |
| }, | |
| "mixup_alpha": 1, | |
| "coef_step": 0, | |
| "num_classes": 10, | |
| "ema_alpha": 0.997, | |
| "ema_update": true, | |
| "ema_step": 40000, | |
| "data_loader": { | |
| "type": "CIFAR10DataLoader", | |
| "args":{ | |
| "data_dir": "dir/to/data", | |
| "batch_size": 128, | |
| "batch_size2": 128, | |
| "num_batches": 0, | |
| "shuffle": true, | |
| "validation_split": 0, | |
| "num_workers": 8, | |
| "pin_memory": true | |
| } | |
| }, | |
| "optimizer1": { | |
| "type": "SGD", | |
| "args":{ | |
| "lr": 0.02, | |
| "momentum": 0.9, | |
| "weight_decay": 5e-4 | |
| } | |
| }, | |
| "optimizer2": { | |
| "type": "SGD", | |
| "args":{ | |
| "lr": 0.02, | |
| "momentum": 0.9, | |
| "weight_decay": 5e-4 | |
| } | |
| }, | |
| "train_loss": { | |
| "type": "elr_plus_loss", | |
| "args":{ | |
| "beta": 0.9, | |
| "lambda": 1 | |
| } | |
| }, | |
| "val_loss": "cross_entropy", | |
| "metrics": [ | |
| "my_metric", "my_metric2" | |
| ], | |
| "lr_scheduler": { | |
| "type": "MultiStepLR", | |
| "args": { | |
| "milestones": [150], | |
| "gamma": 0.1 | |
| } | |
| }, | |
| "trainer": { | |
| "epochs": 200, | |
| "warmup": 0, | |
| "save_dir": "dir/to/model", | |
| "save_period": 1, | |
| "verbosity": 2, | |
| "label_dir": "saved/", | |
| "monitor": "max val_my_metric", | |
| "early_stop": 2000, | |
| "tensorboard": false, | |
| "mlflow": true, | |
| "_percent": "Percentage of noise", | |
| "percent": 0.4, | |
| "_begin": "When to begin updating labels", | |
| "begin": 0, | |
| "_asym": "symmetric noise if false", | |
| "asym": true | |
| } | |
| } | |