Abhishek Tandon
Results
3
comments of
Abhishek Tandon
@szalpal Is there any update on this?
@anhminh3105 Faced the same issue but was able to resolve it by using map_location while loading the weights: ```python checkpoint = torch.load("original_ckpt.t7", map_location='cuda:0') net_dict = checkpoint['net_dict'] net.load_state_dict(net_dict) net.eval() ``` Hope...
Is there any update on this?