model_config { arch: "resnet", n_layers: 18 # Setting these parameters to true to match the template downloaded from NGC. use_batch_norm: true all_projections: true freeze_blocks: 0 freeze_blocks: 1 input_image_size: "3,224,224" } train_config { train_dataset_path: "/home/omno/Desktop/umair/tlt-samples/classification/data/train" val_dataset_path: "/home/omno/Desktop/umair/tlt-samples/classification/data/test" pretrained_model_path: "/home/omno/Desktop/umair/tlt-samples/classification/pretrained_resnet18/tlt_pretrained_classification_vresnet18/resnet_18.hdf5" optimizer { sgd { lr: 0.01 decay: 0.0 momentum: 0.9 nesterov: False } } batch_size_per_gpu: 64 n_epochs: 80 n_workers: 16 preprocess_mode: "caffe" enable_random_crop: True enable_center_crop: True label_smoothing: 0.0 mixup_alpha: 0.1 # regularizer reg_config { type: "L2" scope: "Conv2D,Dense" weight_decay: 0.00005 } # learning_rate lr_config { step { learning_rate: 0.006 step_size: 10 gamma: 0.1 } } } eval_config { eval_dataset_path: "/home/omno/Desktop/umair/tlt-samples/classification/data/test" model_path: "/workspace/tlt-experiments/classification/output/weights/resnet_080.tlt" top_k: 3 batch_size: 256 n_workers: 8 enable_center_crop: True }