Skip to content

Commit

Permalink
Update training_loop.py
Browse files Browse the repository at this point in the history
correcting default snapshot setting for stage1 training.
  • Loading branch information
YuDeng committed Jun 7, 2021
1 parent bf27d09 commit 0530bd1
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions training/training_loop.py
Original file line number Diff line number Diff line change
Expand Up @@ -117,9 +117,9 @@ def training_loop(
network_snapshot_ticks = 10, # How often to export network snapshots?
save_tf_graph = True, # Include full TensorFlow computation graph in the tfevents file?
save_weight_histograms = False, # Include weight histograms in the tfevents file?
resume_run_id = 87, # Run ID or network pkl to resume training from, None = start from scratch.
resume_snapshot = 2364, # Snapshot index to resume training from, None = autodetect.
resume_kimg = 2364, # Assumed training progress at the beginning. Affects reporting and training schedule.
resume_run_id = None, # Run ID or network pkl to resume training from, None = start from scratch.
resume_snapshot = None, # Snapshot index to resume training from, None = autodetect.
resume_kimg = 0.0, # Assumed training progress at the beginning. Affects reporting and training schedule.
resume_time = 0.0,
**_kwargs): # Assumed wallclock time at the beginning. Affects reporting.

Expand Down Expand Up @@ -298,4 +298,4 @@ def training_loop(

ctx.close()

#----------------------------------------------------------------------------
#----------------------------------------------------------------------------

0 comments on commit 0530bd1

Please sign in to comment.