fancy-chef-63734
05/01/2024, 3:20 PMclass TrainFlow(FlowSpec):
use_latest_check_point = Parameter("use_latest_check_point", default=False)
@step
def start(self):
…
@retry(times=1)
@step
def train(self):
if self.retry == 1:
# This step has failed and is now in a retry
self.use_latest_check_point = True
train(use_latest_check_point=self.use_latest_check_point)