PointPWC icon indicating copy to clipboard operation
PointPWC copied to clipboard

Published 20 hours ago • verified publisher icon DylanWusee

[Solution] Unable to continue training model on multi_gpu

Open MaxShi007 opened this issue 2 years ago • 0 comments

model.module.load_state_dict(torch.load(args.pretrain)) https://github.com/DylanWusee/PointPWC/blob/f882e4c396b338d6144bf3d07b335ccaf155bdcf/train.py#L116

MaxShi007 avatar Jun 09 '22 14:06 MaxShi007