Commit 438f76dc authored by Cristian Lalescu's avatar Cristian Lalescu
Browse files

add more general checkpoint file name method

parent 2836c5b6
Pipeline #44483 failed with stage
...@@ -781,6 +781,10 @@ class DNS(_code): ...@@ -781,6 +781,10 @@ class DNS(_code):
return os.path.join( return os.path.join(
self.work_dir, self.work_dir,
self.simname + '_checkpoint_0.h5') self.simname + '_checkpoint_0.h5')
def get_checkpoint_fname(self, checkpoint = 0):
return os.path.join(
self.work_dir,
self.simname + '_checkpoint_{0}.h5'.format(checkpoint))
def generate_tracer_state( def generate_tracer_state(
self, self,
rseed = None, rseed = None,
......
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