add import nn to train script

2018-03-11 21:27:48 +00:00 · 2018-03-11 21:27:48 +00:00 · 2103cf1e91
parent cb0cfa3e19
commit 2103cf1e91
1 changed files with 1 additions and 0 deletions
--- a/train_img_model_xent.py
+++ b/train_img_model_xent.py
@ -8,6 +8,7 @@ import os.path as osp
 import numpy as np

 import torch
+import torch.nn as nn
 import torch.backends.cudnn as cudnn
 from torch.utils.data import DataLoader
 from torch.autograd import Variable