modified: autoencoder2d_defaults.yaml

	modified:   autoencoder3d_defaults.yaml

Author: nkern

py21cmnet/config/autoencoder2d_defaults.yaml

@@ -11,7 +11,7 @@ conv_layer :
     padding_mode : 'circular'
   activation : 'ReLU'
   act_kwargs : {}
-  batch_norm : 'BatchNorm2d'
+  norm : 'BatchNorm2d'
   norm_kwargs : {}
   dropout : None
   dropout_kwargs : {}

py21cmnet/config/autoencoder3d_defaults.yaml

@@ -11,7 +11,7 @@ conv_layer :
     padding_mode : 'circular'
   activation : 'ReLU'
   act_kwargs : {}
-  batch_norm : 'BatchNorm3d'
+  norm : 'BatchNorm3d'
   norm_kwargs : {}
   dropout : None
   dropout_kwargs : {}