Commit ccdb8639 authored by Sebastian Nickels's avatar Sebastian Nickels

Removed fix

parent 55d9f701
Pipeline #226586 failed with stages
in 32 seconds
......@@ -231,7 +231,7 @@ class ${tc.fileNameWithoutEnding}:
begin_epoch = 0
if load_checkpoint:
begin_epoch = self._net_creator.load(mx_context) + 1
begin_epoch = self._net_creator.load(mx_context)
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) + 1
begin_epoch = self._net_creator.load(mx_context)
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) + 1
begin_epoch = self._net_creator.load(mx_context)
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) + 1
begin_epoch = self._net_creator.load(mx_context)
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