mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

add import nn to train script

pull/17/head
KaiyangZhou 2018-03-11 21:27:48 +00:00
parent cb0cfa3e19
commit 2103cf1e91
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train_img_model_xent.py
View File

@ -8,6 +8,7 @@ import os.path as osp
import numpy as np import numpy as np
import torch import torch
import torch.nn as nn
import torch.backends.cudnn as cudnn import torch.backends.cudnn as cudnn
from torch.utils.data import DataLoader from torch.utils.data import DataLoader
from torch.autograd import Variable from torch.autograd import Variable