你的模型或者参数没有放到GPU上,解决:
backbone = eval("backbones.{}".format(args.network))(False, dropout=dropout, fp16=args.fp16).to(rank)
最后的 .to(rank) 做到了这一点。在我的代码中rank=0
你的模型或者参数没有放到GPU上,解决:
backbone = eval("backbones.{}".format(args.network))(False, dropout=dropout, fp16=args.fp16).to(rank)
最后的 .to(rank) 做到了这一点。在我的代码中rank=0