diff --git a/aaa_helpers.txt b/aaa_helpers.txt
index 11f8808..1ca257e 100644
--- a/aaa_helpers.txt
+++ b/aaa_helpers.txt
@@ -1,5 +1,6 @@
 nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_64_e2e_r3" > nohup/review/pyr_abl_32_64_e2e_r3.out 2>&1 &
 nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_64_e2e_r3_pre" > nohup/review/pyr_abl_32_64_e2e_r3_pre.out 2>&1 &
+nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_64_e2e_r3_fine" > nohup/review/pyr_abl_32_64_e2e_r3_fine.out 2>&1 &
 nohup python evaluate_ae_pyramid.py --name evaluate_pyr_abl_32_64_e2e_r3 > nohup/evaluate/pyr_abl_32_64_e2e_r3.out 2>&1 &
 nohup python calculate_error.py --name error_pyr_abl_32_64_e2e_r3 > nohup/error/pyr_abl_32_64_e2e_r3.out 2>&1 &
 
diff --git a/configs/config_abl_pyr_32.py b/configs/config_abl_pyr_32.py
index 5aeef36..195d5d1 100644
--- a/configs/config_abl_pyr_32.py
+++ b/configs/config_abl_pyr_32.py
@@ -1,7 +1,7 @@
 config = dict()
 
 config['gpu'] = '0'
-config['name'] = 'pyr_abl_32_64_e2e_r3_pre'
+config['name'] = 'pyr_abl_32_64_e2e_r3_fine_2'
 config['notes'] = ''
 config['tags'] = ['review_3']
 
@@ -26,7 +26,7 @@ config['pyramid']['resolutions'] = [32, 64]
 config['pyramid']['pretrained_pyramid'] = ''
 config['pyramid']['final_upscale_enhancer'] = 'checkpoints/64_512_2'
 
-config['pretrained_model'] = 'checkpoints/pyr_abl_32_64_e2e_r3'
+config['pretrained_model'] = 'checkpoints/pyr_abl_32_64_e2e_r3_fine'
 config['pretrained_model_iteration'] = ''
 
 config['train_dir'] = '/home/workspace/local-data/train'
@@ -44,7 +44,7 @@ config['smpl_res'] = 14475
 if config['is_cape']:
   config['smpl_res'] = 3638
 
-config['lr'] = 0.0001
+config['lr'] = 0.00001
 config['starting_balancing_coeff'] = 0.0001
 config['reg_loss_coeff'] = 0.1 # set to 0 if don't want to use edge+collision
 config['avg_garment_path'] = ''
@@ -68,7 +68,7 @@ config['pyramid']['latent_dim'] = config['latent_dim']
 config['pyramid']['add_noise'] = True
 config['pyramid']['noise_amount'] = 0.025
 
-config['pyramid']['loss_weights'] = [1,1,1,1,1]
+config['pyramid']['loss_weights'] = [2,1,1,1,1]
 
 for res in config['pyramid']['resolutions']:
   if res != 512:
diff --git a/cvae_pyramid.py b/cvae_pyramid.py
index ffcb62c..e85f8e9 100644
--- a/cvae_pyramid.py
+++ b/cvae_pyramid.py
@@ -807,8 +807,12 @@ if __name__ == '__main__':
         # assert len(config['pyramid']['resolutions']) == 1
         load_model(model, config['pretrained_model'], iteration=config['pretrained_model_iteration'])
 
-    model.trainable = False
-    model.cond_encoder_2.trainable = True
+    model.static_enc.trainable = False
+    model.body_enc.trainable = False
+    model.curv_enc.trainable = False
+
+    # model.trainable = False
+    # model.cond_encoder_2.trainable = True
     # if len(config['pyramid']['resolutions']) == 1:
     # else:
     #     load_model(model, config['pyramid']['pretrained_pyramid'], iteration=(config['pyramid']['pretrained_pyramid_iteration'] if 'pretrained_pyramid_iteration' in config.keys() else ''))