config.yaml

{

    #Model
    "model": "EfficientNet", # [Alexnet, VGG, GoogleNet, ResNet, DenseNet, MobileNet, SqueezeNet, ShuffleNet, EfficientNet, SE-ResNet (not available)]
    "pretrained": True,
    "n_class": 40,

  }