Undefined name: import torch
@@ -8,6 +8,7 @@ import sys
import time
import math
+import torch
import torch.nn as nn
import torch.nn.init as init