{ #Dataset "train_path": "./train.txt", "test_path": "./test.txt", "val_path": "./val.txt", "n_data": 1, #Model "model": "EfficientNet", # [Alexnet, VGG, GoogleNet, ResNet, DenseNet, MobileNet, SqueezeNet, ShuffleNet, EfficientNet, SE-ResNet (not available)] "pretrained": True, "n_class": 40, #Training "B_sz": 4, "Lr": 0.001, "Epoch": 5, "optimizer": "Adam" #[Adam, SGD] }