updated static training

configs/iccvw/ae_static_1.py

@@ -3,12 +3,12 @@ from utils.convnext_utils import CNBlockConfig
 config = dict()
 
 config['gpu'] = '3'
-config['name'] = 'static_iccvw_2'
+config['name'] = 'static_iccvw_4'
 config['notes'] = ' ' 
 config['tags'] = ['torch', 'static'] 
 config['script'] = 'scripts/ae_static_train.py'
 
-config['pretrained_model'] = 'checkpoints/iccvw/static_iccvw_1'
+config['pretrained_model'] = 'checkpoints/iccvw/static_iccvw_3'
 
 config['batch_size'] = 4
 config['train_dir'] = '/data-local/127/cloth3d_processed'
@@ -17,7 +17,7 @@ config['test_dir'] = '/data-net/dataset/test_ready'
 
 config['resolution'] = 512
 config['is_skirt'] = False
-config['dataset_in_memory'] = True
+config['dataset_in_memory'] = False
 config['is_cape'] = False
 config['remove_global_orientation'] = True
 config['device'] = 'cuda'

configs/iccvw/ae_static_cape_f.py

@@ -19,7 +19,7 @@ config['valid_frames'] = None
 
 config['resolution'] = 512
 config['is_skirt'] = False
-config['dataset_in_memory'] = True
+config['dataset_in_memory'] = False
 config['is_cape'] = True
 config['remove_global_orientation'] = True
 config['device'] = 'cuda'
@@ -63,7 +63,6 @@ config['cond_ae']['activation'] = 'tanh'
 
 config['lr'] = 0.0005
 config['cond_ae']['upscale_output'] = False
-config['pretrained_model'] = ''
 
 config['signature'] = { 'id',
                         'frame_nr',

configs/iccvw/ae_static_cape_m.py

@@ -19,7 +19,7 @@ config['valid_frames'] = None
 
 config['resolution'] = 512
 config['is_skirt'] = False
-config['dataset_in_memory'] = True
+config['dataset_in_memory'] = False
 config['is_cape'] = True
 config['remove_global_orientation'] = True
 config['device'] = 'cuda'
@@ -63,7 +63,6 @@ config['cond_ae']['activation'] = 'tanh'
 
 config['lr'] = 0.0005
 config['cond_ae']['upscale_output'] = False
-config['pretrained_model'] = ''
 
 config['signature'] = { 'id',
                         'frame_nr',