resnet condition encoders

This commit is contained in:
2023-02-27 17:41:47 +00:00
parent bb7128f589
commit aa3defbf07
10 changed files with 151 additions and 111 deletions
+8 -8
View File
@@ -23,17 +23,17 @@ config['cond_ae']['is_label'] = False
config['cond_ae']['upscale_output'] = False
config['cond_ae']['latent_dim'] = config['latent_dim']
config['cond_ae']['kernel_size'] = 7
config['cond_ae']['bottleneck_size'] = 8*4 # 8*4
config['cond_ae']['layer_params'] = [3, 4, 23, 4, 3]
# config['cond_ae']['layer_params'] = [3, 4, 23, 3, 3]
config['cond_ae']['activation'] = 'linear'
config['cond_ae']['bottleneck_size'] = 8*2 # 8*4
# config['cond_ae']['layer_params'] = [3, 4, 23, 4, 3]
config['cond_ae']['layer_params'] = [3, 7, 15, 7, 3]
config['cond_ae']['activation'] = 'tanh'
config['gpu'] = '5'
config['lr'] = 0.0001
config['gpu'] = '1'
config['lr'] = 3e-4
config['name'] = 'static_resnet_1'
config['notes'] = 'resnet101 based, one more block'
config['tags'] = ['iccv', 'resnet']
config['notes'] = ''
config['tags'] = ['iccv', 'resnet', 'static']