Commit 7ee015a8 authored by Cristian Lalescu's avatar Cristian Lalescu
Browse files

update checkpoint file name method

parent 438f76dc
Pipeline #44540 failed with stage
......@@ -781,7 +781,8 @@ class DNS(_code):
return os.path.join(
self.work_dir,
self.simname + '_checkpoint_0.h5')
def get_checkpoint_fname(self, checkpoint = 0):
def get_checkpoint_fname(self, iteration = 0):
checkpoint = iteration // self.parameters['checkpoints_per_file']
return os.path.join(
self.work_dir,
self.simname + '_checkpoint_{0}.h5'.format(checkpoint))
......
Supports Markdown
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