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

trying to get resnet working

Browse Source
This commit is contained in:
luni
2023-02-22 10:33:29 +00:00
parent e0a5c9a2e3
commit bb7128f589
4 changed files with 24 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/iccv/config_ae_static.py
+3 -3
View File
@@ -23,14 +23,14 @@ 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'] = 4 # 8*4
config['cond_ae']['layer_params'] = [3, 4, 5, 4, 3]
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['gpu'] = '5'
config['lr'] = 0.0005
config['lr'] = 0.0001
config['name'] = 'static_resnet_1'
config['notes'] = 'resnet101 based, one more block'
config['tags'] = ['iccv', 'resnet']