Closed sybil12 closed 4 years ago
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')
rt, I want to eval at a CPU machine, how to do it?