From 20f10f8fbb1b50a87ba32daf1c34599f0ae33bb3 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Thu, 18 Mar 2021 09:48:47 +0200
Subject: [PATCH] Allow to continue running from previous execution iteration

---
 clearml/task.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/clearml/task.py b/clearml/task.py
index c2de0491..1d3876ee 100644
--- a/clearml/task.py
+++ b/clearml/task.py
@@ -516,6 +516,9 @@ class Task(_Task):
                     cls.__update_master_pid_task(task=task)
                     # make sure we are started
                     task.started(ignore_errors=True)
+                    # continue last iteration if we had any
+                    if task.data.last_iteration:
+                        task.set_initial_iteration(int(task.data.last_iteration) + 1)
                 else:
                     # subprocess should get back the task info
                     task = Task.get_task(task_id=is_sub_process_task_id)