Aufgrund einer Wartung wird GitLab am 19.10. zwischen 8:00 und 9:00 Uhr kurzzeitig nicht zur Verfügung stehen. / Due to maintenance, GitLab will be temporarily unavailable on 19.10. between 8:00 and 9:00 am.

Commit 2a6d82a1 authored by Sebastian N.'s avatar Sebastian N.
Browse files

Fixed counting error when starting with loaded params from another training run

parent 65da4505
Pipeline #226556 failed with stages
in 17 seconds
......@@ -231,7 +231,7 @@ class ${tc.fileNameWithoutEnding}:
begin_epoch = 0
if load_checkpoint:
begin_epoch = self._net_creator.load(mx_context)
begin_epoch = self._net_creator.load(mx_context) + 1
else:
if os.path.isdir(self._net_creator._model_dir_):
shutil.rmtree(self._net_creator._model_dir_)
......
......@@ -230,7 +230,7 @@ class CNNSupervisedTrainer_Alexnet:
begin_epoch = 0
if load_checkpoint:
begin_epoch = self._net_creator.load(mx_context)
begin_epoch = self._net_creator.load(mx_context) + 1
else:
if os.path.isdir(self._net_creator._model_dir_):
shutil.rmtree(self._net_creator._model_dir_)
......
......@@ -230,7 +230,7 @@ class CNNSupervisedTrainer_CifarClassifierNetwork:
begin_epoch = 0
if load_checkpoint:
begin_epoch = self._net_creator.load(mx_context)
begin_epoch = self._net_creator.load(mx_context) + 1
else:
if os.path.isdir(self._net_creator._model_dir_):
shutil.rmtree(self._net_creator._model_dir_)
......
......@@ -230,7 +230,7 @@ class CNNSupervisedTrainer_VGG16:
begin_epoch = 0
if load_checkpoint:
begin_epoch = self._net_creator.load(mx_context)
begin_epoch = self._net_creator.load(mx_context) + 1
else:
if os.path.isdir(self._net_creator._model_dir_):
shutil.rmtree(self._net_creator._model_dir_)
......
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