Commit cb751ac

committed

fix fp16 training

1 parent 9792c54 commit cb751acCopy full SHA for cb751ac

3 files changed

+298

-299

lines changed

tools
- main.py
torchreid/engine/image
- am_softmax.py
- multilabel.py

3 files changed

+298

-299

lines changed

`‎tools/main.py`

-1

-Original file line number
+Diff line change
 import os.path as osp
 import sys
 import time
 -from torch._C import device
 from torch.utils.tensorboard import SummaryWriter
 import torch

Comments

(0)