| { | |
| "batch_size": 16, | |
| "epoch": 20, | |
| "num_classes": 2, | |
| "lam_const": 5.0, | |
| "warmup_epoch": 5, | |
| "warmup_type": "sigmoid", | |
| "T": 10.0, | |
| "k": 1228, | |
| "data_opt": { | |
| "image_size": 224, | |
| "use_crop": false, | |
| "jitter": 0.0, | |
| "from_domain": "all", | |
| "alpha": 1.0 | |
| }, | |
| "networks": { | |
| "encoder": { | |
| "name": "resnet101" | |
| }, | |
| "classifier": { | |
| "name": "base", | |
| "in_dim": 1000, | |
| "num_classes": 2 | |
| } | |
| }, | |
| "optimizer": { | |
| "encoder_optimizer": { | |
| "optim_type": "sgd", | |
| "lr": 0.001, | |
| "momentum": 0.9, | |
| "weight_decay": 0.0005, | |
| "nesterov": true, | |
| "sched_type": "step", | |
| "lr_decay_step": 16, | |
| "lr_decay_rate": 0.1 | |
| }, | |
| "classifier_optimizer": { | |
| "optim_type": "sgd", | |
| "lr": 0.01, | |
| "momentum": 0.9, | |
| "weight_decay": 0.0005, | |
| "nesterov": true, | |
| "sched_type": "step", | |
| "lr_decay_step": 16, | |
| "lr_decay_rate": 0.1 | |
| } | |
| } | |
| } |