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,7 +31,7 @@ config['cond_ae']['decoder_dims'] = config['cond_ae']['encoder_dims']
 config['cond_ae']['activation'] = 'linear'


-config['gpu'] = '5'
+config['gpu'] = '0'
 config['lr'] = 0.0005
 config['name'] = 'static_post_4'
 config['notes'] = 'convnext + convnext, | \