diff --git a/models/Model_SAE/Model.py b/models/Model_SAE/Model.py index 8d05743..756010a 100644 --- a/models/Model_SAE/Model.py +++ b/models/Model_SAE/Model.py @@ -376,7 +376,7 @@ class SAEModel(ModelBase): ar += [ [self.decoder_srcm, 'decoder_srcm.h5'], [self.decoder_dstm, 'decoder_dstm.h5'] ] - self.save_weights_safe(ar, opt_ar) + self.save_weights_safe(ar) #override