luni/cvpr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

resnet condition encoders

Browse Source
This commit is contained in:
luni
2023-02-27 17:41:47 +00:00
parent bb7128f589
commit aa3defbf07
10 changed files with 151 additions and 111 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/iccv/config_ae_body.py
+6 -9
View File
@@ -23,19 +23,16 @@ 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'] = 16
config['cond_ae']['encoder_depths'] = [3, 3, 9, 9, 3]
config['cond_ae']['encoder_dims'] = [16, 32, 64, 128, config['cond_ae']['bottleneck_size']]
config['cond_ae']['decoder_depths'] = config['cond_ae']['encoder_depths']
config['cond_ae']['decoder_dims'] = config['cond_ae']['encoder_dims']
config['cond_ae']['bottleneck_size'] = 16 * 2
config['cond_ae']['layer_params'] = [3, 7, 15, 7, 3]
config['cond_ae']['activation'] = 'tanh'
config['gpu'] = '4'
config['gpu'] = '3'
config['lr'] = 0.0001
config['name'] = 'body_r7_3'
config['notes'] = 'resnet18'
config['tags'] = ['review_7']
config['name'] = 'body_resnet_1'
config['notes'] = ''
config['tags'] = ['iccv', 'resnet']
config['cond_ae']['upscale_output'] = False
config['pretrained_model'] = ''
configs/iccv/config_ae_curv.py
+7 -9
View File
@@ -23,19 +23,17 @@ config['cond_ae']['is_label'] = True
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'] = 40
config['cond_ae']['encoder_depths'] = [3, 3, 9, 9, 3]
config['cond_ae']['encoder_dims'] = [16, 32, 64, 128, config['cond_ae']['bottleneck_size']]
config['cond_ae']['decoder_depths'] = config['cond_ae']['encoder_depths']
config['cond_ae']['decoder_dims'] = config['cond_ae']['encoder_dims']
config['cond_ae']['bottleneck_size'] = 40 * 2
config['cond_ae']['layer_params'] = [3, 7, 15, 7, 3]
config['cond_ae']['activation'] = 'tanh'
config['gpu'] = '3'
config['gpu'] = '2'
config['lr'] = 0.001
config['name'] = 'curv_r7_3'
config['notes'] = 'resnet18'
config['tags'] = ['review_7']
config['name'] = 'curv_resnet_1'
config['notes'] = ''
config['tags'] = ['iccv', 'resnet']
config['cond_ae']['upscale_output'] = False
config['pretrained_model'] = ''
configs/iccv/config_ae_static.py
+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']