more training

2023-02-25 22:46:15 +00:00
parent 90cf267dd8
commit aa96c6eb9c
6 changed files with 234 additions and 32 deletions
@@ -33,14 +33,14 @@ config['cond_ae']['decoder_dims'] = config['cond_ae']['encoder_dims']
 config['cond_ae']['activation'] = 'tanh'


-config['gpu'] = '5'
+config['gpu'] = '3'
 config['lr'] = 0.0005
 config['name'] = 'static_iccv_1_cape_m'
 config['notes'] = ' ' 
 config['tags'] = ['iccv', 'static', 'cape_m'] 

 config['cond_ae']['upscale_output'] = False
-config['pretrained_model'] = ''
+config['pretrained_model'] = 'checkpoints/iccv/static_iccv_1_cape_m_old'


 config['signature_dynamic'] = { 'id': (), 'frame_nr': (), }