add report codes

hyunwoongko · Dec 18, 2019 · ef96e13 · ef96e13
1 parent 09c74b8
commit ef96e13
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 2 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/__pycache__/train.cpython-36.pyc b/__pycache__/train.cpython-36.pyc
diff --git a/models/layers/__pycache__/scale_dot_product_attention.cpython-36.pyc b/models/layers/__pycache__/scale_dot_product_attention.cpython-36.pyc
diff --git a/result/__init__.py b/result/__init__.py
@@ -0,0 +1,5 @@
+"""
+@author : Hyunwoong
+@when : 2019-12-18
+@homepage : https://github.com/gusdnd852
+"""
diff --git a/train.py b/train.py
@@ -87,18 +87,29 @@ def evaluate(model, iterator, criterion):
 
 
 def run(total_epoch, best_loss):
+ train_losses, test_losses = [], []
  for step in range(total_epoch):
  start_time = time.time()
  train_loss = train(model, train_iter, optimizer, criterion, clip)
  valid_loss = evaluate(model, valid_iter, criterion)
  end_time = time.time()
 
+ train_losses.append(train_loss)
+ test_losses.append(valid_loss)
  epoch_mins, epoch_secs = epoch_time(start_time, end_time)
 
  if valid_loss < best_loss:
  best_loss = valid_loss
  torch.save(model.state_dict(), 'saved/model-{0}.pt'.format(valid_loss))
 
+ f = open('result/train.txt', 'w')
+ f.write(str(train_losses))
+ f.close()
+
+ f = open('result/test.txt', 'w')
+ f.write(str(test_losses))
+ f.close()
+
  print(f'Epoch: {step + 1} | Time: {epoch_mins}m {epoch_secs}s')
  print(f'\tTrain Loss: {train_loss:.3f} | Train PPL: {math.exp(train_loss):7.3f}')
  print(f'\t Val. Loss: {valid_loss:.3f} | Val. PPL: {math.exp(valid_loss):7.3f}')