WayneLin015
Results
1
comments of
WayneLin015
@YaLTeR add pytorch_model.cpu() after pytorch_model = model_zoo.get(cfg_name, trained=True)