Skip to content

Commit

Permalink
Merge pull request #2 from swarajpande4/feature
Browse files Browse the repository at this point in the history
Fixed epochs and sample_interval
  • Loading branch information
swarajpande5 authored May 24, 2023
2 parents 24091d9 + 7431386 commit e7b8d4b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion code/train_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ def train_model():

start_time = time.time()
acgan = ACGAN()
acgan.train(epochs=120, batch_size=32, sample_interval=10)
acgan.train(epochs=125, batch_size=32, sample_interval=25)
end_time = time.time()

with open('./metrics/train_metric.json', 'w') as f:
Expand Down

0 comments on commit e7b8d4b

Please sign in to comment.