Skip to content

Commit b90cb20

Browse files
authored
Update data.py
1 parent 4790126 commit b90cb20

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

data.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,13 +9,13 @@ class FLAGS(object):
99
def __init__(self):
1010
self.n_epoch = 25 # "Epoch to train [25]"
1111
self.z_dim = 100 # "Num of noise value]"
12-
self.learning_rate = 0.0002 # "Learning rate of for adam [0.0002]")
12+
self.lr = 0.0002 # "Learning rate of for adam [0.0002]")
1313
self.beta1 = 0.5 # "Momentum term of adam [0.5]")
1414
self.batch_size = 64 # "The number of batch images [64]")
1515
self.output_size = 64 # "The size of the output images to produce [64]")
1616
self.sample_size = 64 # "The number of sample images [64]")
1717
self.c_dim = 3 # "Number of image channels. [3]")
18-
self.save_step = 500 # "The interval of saveing checkpoints. [500]")
18+
self.save_every_epoch = 1 # "The interval of saveing checkpoints.")
1919
# self.dataset = "celebA" # "The name of dataset [celebA, mnist, lsun]")
2020
self.checkpoint_dir = "checkpoint" # "Directory name to save the checkpoints [checkpoint]")
2121
self.sample_dir = "samples" # "Directory name to save the image samples [samples]")

0 commit comments

Comments
 (0)