Commit 07fc45c6 authored by Seema Mirchandaney's avatar Seema Mirchandaney
Browse files

minor fix

parent 08cd2c9a
......@@ -52,7 +52,7 @@ def main():
pixel_index,
slices, slices_p) = get_data(ds)
logger.log(f"Loaded in {timer.lap():.2f}s.")
curr_gen = 0
if settings.load_gen > 0: # Load input from previous generation
curr_gen = settings.load_gen
phased, orientations, orientations_p = checkpoint.load_checkpoint(settings.out_dir, settings.load_gen)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment