add extra block, lower feature sizes

2023-01-25 23:10:51 +00:00
parent 46e37e22eb
commit 9cd9c6f12a
6 changed files with 30 additions and 19 deletions
@@ -31,10 +31,10 @@ config['cond_ae']['decoder_dims'] = config['cond_ae']['encoder_dims']
 config['cond_ae']['activation'] = 'linear'


-config['gpu'] = '0'
+config['gpu'] = '6'
 config['lr'] = 0.0005
-config['name'] = 'static_r7_2'
-config['notes'] = 'resnet18 based'
+config['name'] = 'static_r7_3'
+config['notes'] = 'resnet18 based, one more block'
 config['tags'] = ['review_7']