-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
39 lines (38 loc) · 897 Bytes
/
config.py
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
config = {
'device' : {
'home directory' : './',
'initial seed': 1,
'device type': 'cpu',
},
'data' : {
'batch size': 32,
'num workers': 8,
'image shape': (1, 28, 28),
'train samples': 60000,
'test samples': 4000
},
'generator' : {
'generator blocks' : 3,
},
'discriminator' : {
'discriminator blocks' : 3,
},
'hyperparameters' : {
'latent dimension': 100,
'learning rate': 0.0002,
'beta1': 0.5,
'beta2': 0.999,
'epochs': 200,
'discriminator epochs': 1,
'generator epochs': 1,
},
'save' : {
'sample interval': 1,
'sample save path': 'samples',
'model save path': 'weights'
},
'log' : {
'log path': 'logs',
'experiment number': 1
}
}