model_config { arch: "resnet", n_layers: 18 use_bias: true use_batch_norm: true use_pooling: true all_projections: true freeze_blocks: 0 freeze_blocks: 1 freeze_bn: false freeze_blocks: 2 input_image_size: "3,96,96" batch_norm_config { momentum: 0.9 epsilon: 0.00005 } activation { activation_type: 'relu' } } train_config { train_dataset_path: "/dataset/train" val_dataset_path: "/dataset/val" pretrained_model_path: "/pretrained/resnet_18.hdf5" optimizer: { sgd { lr: 0.006 decay: 0 momentum: 0.9 nesterov: False } } batch_size_per_gpu: 32 n_epochs: 200 n_workers: 16 reg_config { type: "L2" scope: "Conv2D,Dense" weight_decay: 0.000015 } lr_config { cosine { learning_rate: 0.04 soft_start: 0.0 } } random_seed: 2022 enable_random_crop: False enable_center_crop: False enable_color_augmentation: True mixup_alpha: 0.2 label_smoothing: 0 preprocess_mode: "caffe" image_mean { key: 'b' value: 103.9 } image_mean { key: 'g' value: 116.8 } image_mean { key: 'r' value: 123.7 } } eval_config { eval_dataset_path: "/dataset/val" model_path: "/experiment/unpruned/weights/resnet_200.tlt" top_k: 3 batch_size: 256 n_workers: 8 }