import load_pretrained_weights in tools/visactmap

pull/294/head
KaiyangZhou 2019-11-27 17:04:58 +00:00
parent 67f56ca361
commit 63f357a0ba
1 changed files with 1 additions and 1 deletions

View File

@ -14,7 +14,7 @@ import torch
from torch.nn import functional as F
import torchreid
from torchreid.utils import mkdir_if_missing, check_isfile
from torchreid.utils import mkdir_if_missing, check_isfile, load_pretrained_weights
IMAGENET_MEAN = [0.485, 0.456, 0.406]