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 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