Final Train Runs

models/cnn_baseline.py

@@ -15,7 +15,9 @@ class CNNBaseline(CombinedModelMixins,
                   ):
 
     def __init__(self, in_shape, n_classes, weight_init, activation, use_bias, use_norm, dropout, lat_dim, features,
-                 filters, lr, weight_decay, sto_weight_avg, lr_warm_restart_epochs, opt_reset_interval, loss):
+                 filters,
+                 lr, weight_decay, sto_weight_avg, lr_warm_restart_epochs, opt_reset_interval,
+                 loss, scheduler):
 
         # TODO: Move this to parent class, or make it much easieer to access....
         a = dict(locals())