mirrors/deep-person-reid
1
0
Fork 0
mirror of https://github.com/KaiyangZhou/deep-person-reid.git synced 2025-06-03 14:53:23 +08:00
Code Issues Projects Releases Wiki Activity

add import nn to train script

Browse Source
This commit is contained in:
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 torch
import torch.nn as nn
import torch.backends.cudnn as cudnn
from torch.utils.data import DataLoader
from torch.autograd import Variable