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

resnet 101 working

Browse Source
This commit is contained in:
luni
2023-02-21 22:30:32 +00:00
parent 8447b7c505
commit 9eb1183df4
4 changed files with 15 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/iccv/config_ae_static.py
+4 -6
View File
@@ -1,7 +1,7 @@
config = dict()
config['batch_size'] = 4
config['batch_size'] = 4 ##################################### CHANGE CHANGE CHANGE CHANGE CHANGE CHANGE CHANGE
config['train_dir'] = '/home/workspace/local-data/train'
config['valid_dir'] = '/home/workspace/local-data/valid'
config['test_dir'] = '/home/workspace/data/dataset/test_ready'
@@ -23,11 +23,9 @@ 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
config['cond_ae']['encoder_depths'] = [3, 3, 3, 27, 3]
config['cond_ae']['encoder_dims'] = [8, 64, 128, 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'] = 4 # 8*4
config['cond_ae']['layer_params'] = [3, 4, 5, 4, 3]
# config['cond_ae']['layer_params'] = [3, 4, 23, 3, 3]
config['cond_ae']['activation'] = 'linear'