Advertisement
Guest User

Untitled

a guest
Mar 25th, 2019
225
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 2.19 KB | None | 0 0
  1.  
  2. The following have been reloaded with a version change:
  3. 1) cuDNN/7.4.1-CUDA-9.0.176 => cuDNN/7.0.5-CUDA-9.0.176
  4.  
  5.  
  6. The following have been reloaded with a version change:
  7. 1) CUDA/9.0.176 => CUDA/9.1.85
  8. 2) cuDNN/7.0.5-CUDA-9.0.176 => cuDNN/7.0.5-CUDA-9.1.85
  9.  
  10. [2019-03-25 13:14:32,473 INFO] Loading checkpoint from models/tedpure_step_73600.pt
  11. [2019-03-25 13:14:39,412 INFO] Loading vocab from checkpoint at models/tedpure_step_73600.pt.
  12. [2019-03-25 13:14:39,418 INFO] * src vocab size = 12245
  13. [2019-03-25 13:14:39,428 INFO] * tgt vocab size = 15721
  14. [2019-03-25 13:14:39,438 INFO] Building model...
  15. [2019-03-25 13:14:53,643 INFO] NMTModel(
  16. (encoder): RNNEncoder(
  17. (embeddings): Embeddings(
  18. (make_embedding): Sequential(
  19. (emb_luts): Elementwise(
  20. (0): Embedding(12245, 512, padding_idx=1)
  21. )
  22. )
  23. )
  24. (rnn): LSTM(512, 256, num_layers=3, dropout=0.2, bidirectional=True)
  25. )
  26. (decoder): InputFeedRNNDecoder(
  27. (embeddings): Embeddings(
  28. (make_embedding): Sequential(
  29. (emb_luts): Elementwise(
  30. (0): Embedding(15721, 512, padding_idx=1)
  31. )
  32. )
  33. )
  34. (dropout): Dropout(p=0.2)
  35. (rnn): StackedLSTM(
  36. (dropout): Dropout(p=0.2)
  37. (layers): ModuleList(
  38. (0): LSTMCell(1024, 512)
  39. (1): LSTMCell(512, 512)
  40. (2): LSTMCell(512, 512)
  41. )
  42. )
  43. (attn): GlobalAttention(
  44. (linear_context): Linear(in_features=512, out_features=512, bias=False)
  45. (linear_query): Linear(in_features=512, out_features=512, bias=True)
  46. (v): Linear(in_features=512, out_features=1, bias=False)
  47. (linear_out): Linear(in_features=1024, out_features=512, bias=True)
  48. (linear_cover): Linear(in_features=1, out_features=512, bias=False)
  49. )
  50. )
  51. (generator): Sequential(
  52. (0): Linear(in_features=512, out_features=15721, bias=True)
  53. (1): LogSoftmax()
  54. )
  55. )
  56. [2019-03-25 13:14:53,648 INFO] encoder: 11000320
  57. [2019-03-25 13:14:53,651 INFO] decoder: 24516969
  58. [2019-03-25 13:14:53,653 INFO] * number of parameters: 35517289
  59. [2019-03-25 13:14:54,210 INFO] Starting training on GPU: [0]
  60. [2019-03-25 13:14:54,211 INFO] Start training loop and validate every 50 steps...
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement