bug in metric calculation

2021-03-27 16:39:07 +01:00
parent 82835295a1
commit 37e36df0a8
12 changed files with 967 additions and 239 deletions
--- a/models/cnn_baseline.py
+++ b/models/cnn_baseline.py
@ -16,8 +16,7 @@ class CNNBaseline(CombinedModelMixins,
                  ):

    def __init__(self, in_shape, n_classes, weight_init, activation,
-                 use_bias, use_norm, dropout, lat_dim, features,
-                 filters,
+                 use_bias, use_norm, dropout, lat_dim, filters,
                 lr, weight_decay, sto_weight_avg, lr_warm_restart_epochs, opt_reset_interval,
                 loss, scheduler, lr_scheduler_parameter
                 ):