ClassifierTrainingGAN/old_params.py at master · HabibSlim/ClassifierTrainingGAN · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
''' Parameters
   General architecture used for the GAN models '''

params = {'dataset': 'C10',
          'augment': False,
          'num_workers': 8,
          'pin_memory': True,
          'shuffle': True,
          'load_in_mem': False,
          'use_multiepoch_sampler': False,
          'G_param': 'SN',
          'D_param': 'SN',
          'G_ch': 64,
          'D_ch': 64,
          'G_depth': 1,
          'D_depth': 1,
          'D_wide': True,
          'G_shared': False,
          'shared_dim': 0,
          'dim_z': 128,
          'z_var': 1.0,
          'hier': False,
          'cross_replica': False,
          'mybn': False,
          'G_nl': 'relu',
          'D_nl': 'relu',
          'G_attn': '0',
          'D_attn': '0',
          'norm_style': 'bn',
          'seed': 0,
          'G_init': 'N02',
          'D_init': 'N02',
          'skip_init': False,
          'G_lr': 0.0002,
          'D_lr': 0.0002,
          'G_B1': 0.0,
          'D_B1': 0.0,
          'G_B2': 0.999,
          'D_B2': 0.999,
          'batch_size': 512,
          'G_batch_size': 0,
          'num_G_accumulations': 1,
          'num_D_steps': 4,
          'num_D_accumulations': 1,
          'split_D': False,
          'num_epochs': 1000,
          'parallel': True,
          'G_fp16': False,
          'D_fp16': False,
          'D_mixed_precision': False,
          'G_mixed_precision': False,
          'accumulate_stats': False,
          'num_standing_accumulations': 16,
          'G_eval_mode': False,
          'save_every': 500,
          'num_save_copies': 2,
          'num_best_copies': 5,
          'which_best': 'IS',
          'no_fid': False,
          'test_every': 500,
          'num_inception_images': 50000,
          'hashname': False,
          'base_root': '',
          'data_root': 'data',
          'weights_root': 'weights',
          'logs_root': 'logs',
          'samples_root': 'samples',
          'pbar': 'mine',
          'name_suffix': '',
          'experiment_name': '',
          'config_from_name': False,
          'ema': True,
          'ema_decay': 0.9999,
          'use_ema': True,
          'ema_start': 1000,
          'adam_eps': 1e-08,
          'BN_eps': 1e-05,
          'SN_eps': 1e-08,
          'num_G_SVs': 1,
          'num_D_SVs': 1,
          'num_G_SV_itrs': 1,
          'num_D_SV_itrs': 1,
          'G_ortho': 0.0,
          'D_ortho': 0.0,
          'toggle_grads': True,
          'which_train_fn': 'GAN',
          'load_weights': '',
          'resume': False,
          'logstyle': '%3.3e',
          'log_G_spectra': False,
          'log_D_spectra': False,
          'sv_log_interval': 10,
          'sample_npz': True,
          'sample_num_npz': 1,
          'sample_sheets': False,
          'sample_interps': False,
          'sample_sheet_folder_num': -1,
          'sample_random': False,
          'sample_trunc_curves': '0.05_0.05_1.0',
          'sample_inception_metrics': False}