Spaces:
Configuration error
Configuration error
| { | |
| "name": "Webvision_ELR_plus_InceptionResNetV2", | |
| "n_gpu": 2, | |
| "seed": 123, | |
| "arch": { | |
| "args": {"num_classes":50} | |
| }, | |
| "arch1": { | |
| "type": "InceptionResNetV2", | |
| "args": {"num_classes":50} | |
| }, | |
| "arch2": { | |
| "type": "InceptionResNetV2", | |
| "args": {"num_classes":50} | |
| }, | |
| "mixup_alpha": 1.5, | |
| "mixup_ramp": false, | |
| "num_classes": 50, | |
| "ema_alpha": 0.997, | |
| "ema_update": false, | |
| "ema_step": 40000, | |
| "data_loader": { | |
| "type": "WebvisionDataLoader", | |
| "args":{ | |
| "data_dir": "/dir/to/data", | |
| "batch_size": 32, | |
| "batch_size2": 32, | |
| "shuffle": true, | |
| "num_batches": 0, | |
| "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.7, | |
| "lambda": 3 | |
| } | |
| }, | |
| "val_loss": "cross_entropy", | |
| "metrics": [ | |
| "my_metric", "my_metric2" | |
| ], | |
| "lr_scheduler": { | |
| "type": "MultiStepLR", | |
| "args": { | |
| "milestones": [50], | |
| "gamma": 0.1 | |
| } | |
| }, | |
| "trainer": { | |
| "epochs": 100, | |
| "warmup": 0, | |
| "save_dir": "/dir/to/data", | |
| "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.9, | |
| "_begin": "When to begin updating labels", | |
| "begin": 0, | |
| "_asym": "symmetric noise if false", | |
| "asym": false | |
| } | |
| } | |