diff --git a/models/Model_SAE/Model.py b/models/Model_SAE/Model.py index 0429cda..c8ed479 100644 --- a/models/Model_SAE/Model.py +++ b/models/Model_SAE/Model.py @@ -124,7 +124,7 @@ class SAEModel(ModelBase): target_dst_ar = [ Input ( ( bgr_shape[0] // (2**i) ,)*2 + (bgr_shape[-1],) ) for i in range(ms_count-1, -1, -1)] target_dstm_ar = [ Input ( ( mask_shape[0] // (2**i) ,)*2 + (mask_shape[-1],) ) for i in range(ms_count-1, -1, -1)] - use_bn = True + use_bn = False models_list = [] weights_to_load = [] @@ -791,4 +791,4 @@ Model = SAEModel # b = x[:,1] # idxs = (x[:,0][ np.argwhere ( b [ b > (np.mean(b)+np.std(b)) ] )[:,0] ]).astype(np.uint) # generators_list[1].repeat_sample_idxs(idxs) #ask generator to repeat these sample idxs -# print ("dst repeated %d" % (len(idxs)) ) +# print ("dst repeated %d" % (len(idxs)) ) \ No newline at end of file