diff --git a/configs/iccvw/ae_norm_1.py b/configs/iccvw/ae_norm_1.py
index f4cdb16..c9c1383 100644
--- a/configs/iccvw/ae_norm_1.py
+++ b/configs/iccvw/ae_norm_1.py
@@ -58,7 +58,6 @@ block_setting_reverse = [
     CNBlockConfig(64, 32, 9),
     CNBlockConfig(32, 16, 3),
     CNBlockConfig(16, 3, 3),
-    # CNBlockConfig(3, 3, 3),
 ]
 
 config['cond_ae']['encoder_block_settings'] = block_setting
diff --git a/scripts/ae_norm_train.py b/scripts/ae_norm_train.py
index 597e662..af79d1c 100644
--- a/scripts/ae_norm_train.py
+++ b/scripts/ae_norm_train.py
@@ -205,15 +205,15 @@ if __name__ == '__main__':
         load_model(model.encoder, config['pretrained_model'] + '_enc')
         load_model(model.decoder, config['pretrained_model'] + '_dec')
 
-    ckpt_saver_enc = CheckpointSaver('checkpoints/iccv/torch/' + config['name'] + '_enc', top_n=5)
-    ckpt_saver_dec = CheckpointSaver('checkpoints/iccv/torch/' + config['name'] + '_dec', top_n=5)
+    ckpt_saver_enc = CheckpointSaver('checkpoints/iccvw/' + config['name'] + '_enc', top_n=5)
+    ckpt_saver_dec = CheckpointSaver('checkpoints/iccvw/' + config['name'] + '_dec', top_n=5)
 
     run_name = config['name']
     print(f'started training: {run_name}')
 
     if not args['debug']:
         # model = torch.compile(model)
-        wandb.watch(model, log='all', log_freq=10000, log_graph=True)
+        wandb.watch(model, log='all', log_freq=20000, log_graph=True)
 
     train()