Skip to content

feat(template): add a gan template #22

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 12 commits into from
Mar 24, 2021
Merged

feat(template): add a gan template #22

Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
b57cc16
add gan
Mar 20, 2021
7941323
Merge remote-tracking branch 'origin/master' into dcgan
Mar 20, 2021
9447815
port dcgan from ignite
Mar 20, 2021
73b1fab
chore: explicit configs and integration test
Mar 23, 2021
2ee3062
fix: config datase
Mar 23, 2021
dcfec33
fix: add missing config
Mar 23, 2021
23698ff
fix: hidden save_* config
Mar 23, 2021
7663c9d
fix: use pathlib / instead os.path.join
Mar 23, 2021
74d91f7
fix: lower log_train
Mar 23, 2021
9f7e14b
fix: \ needed
Mar 23, 2021
7cebdf9
test: add test for gan
Mar 23, 2021
5b25ced
fix: test imports
Mar 23, 2021
File filter

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix: use pathlib / instead os.path.join
  • Loading branch information
ydcjeff committed Mar 23, 2021
commit 7663c9ddb893c6fb13ac448dff2187967006219d
12 changes: 6 additions & 6 deletions templates/gan/main.py.jinja
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -98,7 +98,7 @@ def run(

@trainer.on(Events.ITERATION_COMPLETED(every=config.log_train))
def print_logs(engine):
fname = os.path.join(config.filepath, LOGS_FNAME)
fname = config.filepath / LOGS_FNAME
columns = ["iteration",] + list(engine.state.metrics.keys())
values = [str(engine.state.iteration),] + [str(round(value, 5)) for value in engine.state.metrics.values()]

Expand All @@ -118,7 +118,7 @@ def run(
@trainer.on(Events.EPOCH_COMPLETED)
def save_fake_example(engine):
fake = netG(fixed_noise)
path = os.path.join(config.filepath, FAKE_IMG_FNAME.format(engine.state.epoch))
path = config.filepath / (FAKE_IMG_FNAME.format(engine.state.epoch))
vutils.save_image(fake.detach(), path, normalize=True)

# --------------------------------------------------
Expand All @@ -127,7 +127,7 @@ def run(
@trainer.on(Events.EPOCH_COMPLETED)
def save_real_example(engine):
img, y = engine.state.batch
path = os.path.join(config.filepath, REAL_IMG_FNAME.format(engine.state.epoch))
path = config.filepath / (REAL_IMG_FNAME.format(engine.state.epoch))
vutils.save_image(img, path, normalize=True)

# -------------------------------------------------------------
Expand Down Expand Up @@ -173,11 +173,11 @@ def run(
warnings.warn("Loss plots will not be generated -- pandas or matplotlib not found")

else:
df = pd.read_csv(os.path.join(config.filepath, LOGS_FNAME), delimiter="\t", index_col="iteration")
df = pd.read_csv(config.filepath / LOGS_FNAME, delimiter="\t", index_col="iteration")
_ = df.plot(subplots=True, figsize=(20, 20))
_ = plt.xlabel("Iteration number")
fig = plt.gcf()
path = os.path.join(config.filepath, PLOT_FNAME)
path = config.filepath / PLOT_FNAME

fig.savefig(path)

Expand All @@ -199,7 +199,7 @@ def run(
# ---------------------------------------------
# Setup is done. Now let's run the training
# ---------------------------------------------
trainer.run(loader, config.max_epochs, config.epoch_length)
trainer.run(loader, max_epochs=config.max_epochs, epoch_length=config.epoch_length)


{% block main_fn %}
Expand Down