Commit ccdb8639 authored by Sebastian Nickels's avatar Sebastian Nickels
Browse files

Removed fix

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