Skip to content
This repository was archived by the owner on Nov 21, 2023. It is now read-only.

Commit 6e63f1d

Browse files
author
AFAgarap
committed
Removes the tf.train.Saver().save(), causes to save the wrong checkpoint
1 parent cea5c9a commit 6e63f1d

File tree

3 files changed

+0
-6
lines changed

3 files changed

+0
-6
lines changed

models/gru_softmax/gru_softmax.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -271,8 +271,6 @@ def train(self, checkpoint_path, log_path, model_name, epochs, train_data, train
271271

272272
print('EOF -- Testing done at step {}'.format(step))
273273

274-
saver.save(sess=sess, save_path=os.path.join(checkpoint_path, model_name), global_step=step)
275-
276274
@staticmethod
277275
def predict(batch_size, cell_size, dropout_rate, num_classes, test_data, test_size, checkpoint_path, result_path):
278276
"""Classifies the data whether there is an intrusion or none

models/gru_svm/gru_svm.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -276,8 +276,6 @@ def train(self, checkpoint_path, log_path, model_name, epochs, train_data, train
276276

277277
print('EOF -- Testing done at step {}'.format(step))
278278

279-
saver.save(sess=sess, save_path=os.path.join(checkpoint_path, model_name), global_step=step)
280-
281279
@staticmethod
282280
def predict(batch_size, cell_size, dropout_rate, num_classes, test_data, test_size, checkpoint_path, result_path):
283281
"""Classifies the data whether there is an intrusion or none

models/svm/svm.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -242,8 +242,6 @@ def train(self, checkpoint_path, log_path, model_name, epochs, result_path, trai
242242

243243
print('EOF -- Testing done at step {}'.format(step))
244244

245-
saver.save(sess, checkpoint_path + model_name, global_step=step)
246-
247245
@staticmethod
248246
def predict(batch_size, num_classes, test_data, test_size, checkpoint_path, result_path):
249247
"""Classifies the data whether there is an intrusion or none

0 commit comments

Comments
 (0)