Lightweight models for real-time semantic segmentation(include mobilenetv1-v3, shufflenetv1-v2, igcv3, efficientnet).
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool
sybil12 opened this issue 5 years ago · comments
rt, I want to eval at a CPU machine, how to do it?
Please change this to model.load_state_dict(torch.load(get_model_file('mobilenet_%s_best_model' % (acronyms[dataset]), root=root)), map_location='cpu') if you want to use mobilenet.
model.load_state_dict(torch.load(get_model_file('mobilenet_%s_best_model' % (acronyms[dataset]), root=root)), map_location='cpu')
