Browse Source

skip training while n_epoch in trainer is not greater than 0

tags/v0.3.1^2
xuyige 6 years ago
parent
commit
e0d6a259ae
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      fastNLP/core/trainer.py

+ 4
- 0
fastNLP/core/trainer.py View File

@@ -181,6 +181,10 @@ class Trainer(object):

"""
results = {}
if self.n_epochs <= 0:
print(f"training epoch is {self.n_epochs}, nothing was done.")
results['seconds'] = 0.
return results
try:
if torch.cuda.is_available() and self.use_cuda:
self.model = self.model.cuda()


Loading…
Cancel
Save