Commit cd626bc9 authored by Sebastian Nickels's avatar Sebastian Nickels

Fixed epoch bug for real

parent 160519ea
Pipeline #226597 failed with stages
in 20 seconds
......@@ -404,7 +404,7 @@ class ${tc.fileNameWithoutEnding}:
network.save_parameters(self.parameter_path(i) + '-' + str(epoch).zfill(4) + '.params')
for i, network in self._networks.items():
network.save_parameters(self.parameter_path(i) + '-' + str(num_epoch + begin_epoch).zfill(4) + '.params')
network.save_parameters(self.parameter_path(i) + '-' + str(num_epoch + begin_epoch + 1).zfill(4) + '.params')
network.export(self.parameter_path(i) + '_newest', epoch=0)
def parameter_path(self, index):
......
......@@ -523,7 +523,7 @@ class CNNSupervisedTrainer_Alexnet:
network.save_parameters(self.parameter_path(i) + '-' + str(epoch).zfill(4) + '.params')
for i, network in self._networks.items():
network.save_parameters(self.parameter_path(i) + '-' + str(num_epoch + begin_epoch).zfill(4) + '.params')
network.save_parameters(self.parameter_path(i) + '-' + str(num_epoch + begin_epoch + 1).zfill(4) + '.params')
network.export(self.parameter_path(i) + '_newest', epoch=0)
def parameter_path(self, index):
......
......@@ -523,7 +523,7 @@ class CNNSupervisedTrainer_CifarClassifierNetwork:
network.save_parameters(self.parameter_path(i) + '-' + str(epoch).zfill(4) + '.params')
for i, network in self._networks.items():
network.save_parameters(self.parameter_path(i) + '-' + str(num_epoch + begin_epoch).zfill(4) + '.params')
network.save_parameters(self.parameter_path(i) + '-' + str(num_epoch + begin_epoch + 1).zfill(4) + '.params')
network.export(self.parameter_path(i) + '_newest', epoch=0)
def parameter_path(self, index):
......
......@@ -523,7 +523,7 @@ class CNNSupervisedTrainer_VGG16:
network.save_parameters(self.parameter_path(i) + '-' + str(epoch).zfill(4) + '.params')
for i, network in self._networks.items():
network.save_parameters(self.parameter_path(i) + '-' + str(num_epoch + begin_epoch).zfill(4) + '.params')
network.save_parameters(self.parameter_path(i) + '-' + str(num_epoch + begin_epoch + 1).zfill(4) + '.params')
network.export(self.parameter_path(i) + '_newest', epoch=0)
def parameter_path(self, index):
......
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