a b/configs/training/EfficientNetB4.json 

  1
  
{





  
  

  2
  
  "exp_name":"exp-efficientnet",





  
  

  3
  
  "type": "Trainer2D",





  
  

  4
  
  "device": "cpu",





  
  

  5
  
  "model": "EfficientNetB4",





  
  

  6
  





  
  

  7
  
  "train_json": "data/train.json",





  
  

  8
  
  "val_json": "data/val.json",





  
  

  9
  
  "mapping_json": "data/class-mapper.json",





  
  

  10
  
  "num_classes": 8,





  
  

  11
  





  
  

  12
  
  "batch_size": 32,





  
  

  13
  
  "optim": "Adam",





  
  

  14
  
  "optim_params":





  
  

  15
  
  {





  
  

  16
  
    "lr": 1e-3





  
  

  17
  
  },





  
  

  18
  





  
  

  19
  
  "num_workers": 3,





  
  

  20
  
  "exp_dir" : "experiments/",





  
  

  21
  





  
  

  22
  
  "model_path": 0,





  
  

  23
  
  "epochs": 1





  
  

  24
  
}