HemanM commited on
Commit
a03a6ca
·
verified ·
1 Parent(s): 84c0256

Update evo_model.py

Browse files
Files changed (1) hide show
  1. evo_model.py +3 -0
evo_model.py CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  import torch.nn as nn
2
  from torch.nn import TransformerEncoder, TransformerEncoderLayer
3
 
 
1
+
2
+ import torch
3
+
4
  import torch.nn as nn
5
  from torch.nn import TransformerEncoder, TransformerEncoderLayer
6