File size: 1,250 Bytes
72b128c |
1 |
{"model.arch": "resnet50", "model.pretrained": 0, "model.single_head": 0, "model.multi_head": 0, "model.nesting_start": 3, "model.fixed_feature": 1024, "resolution.min_res": 160, "resolution.max_res": 192, "resolution.end_ramp": 34, "resolution.start_ramp": 29, "data.train_dataset": "/home/jupyter/dataset-ffcv/train_500_0.50_90.ffcv", "data.val_dataset": "/home/jupyter/dataset-ffcv/uncompressed/val_500_uncompressed.ffcv", "data.num_workers": 12, "data.in_memory": 1, "lr.step_ratio": 0.1, "lr.step_length": 30, "lr.lr_schedule_type": "cyclic", "lr.lr": 0.425, "lr.lr_peak_epoch": 2, "logging.folder": "trainlogs", "logging.log_level": 1, "validation.batch_size": 512, "validation.resolution": 256, "validation.lr_tta": 1, "validation.model_weights_disk": "", "validation.bias_zero": 0, "training.eval_only": 0, "training.batch_size": 512, "training.optimizer": "sgd", "training.momentum": 0.9, "training.weight_decay": 0.0001, "training.epochs": 40, "training.label_smoothing": 0.1, "training.distributed": 1, "training.use_blurpool": 1, "training.resume": "checkpoints/checkpoint.pth.tar", "training.tensorboard": 1, "training.lincls": 2048, "training.pretrain_model": "", "dist.world_size": 2, "dist.address": "localhost", "dist.port": "12355"} |