fixed feature sizes, cvae working

2023-01-25 21:00:47 +00:00
parent cba9d8ce90
commit 46e37e22eb
12 changed files with 39 additions and 43 deletions
@@ -31,9 +31,9 @@ config['cond_ae']['decoder_dims'] = config['cond_ae']['encoder_dims']
 config['cond_ae']['activation'] = 'linear'


-config['gpu'] = '6'
+config['gpu'] = '0'
 config['lr'] = 0.0005
-config['name'] = 'static_r7'
+config['name'] = 'static_r7_2'
 config['notes'] = 'resnet18 based'
 config['tags'] = ['review_7']