config.yaml

{
    #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]
  }