Closed eps696 closed 5 years ago
you don't have to unzip the tarfile for loading weights, "torch.load" loads weights from tarfile itself. like
checkpoint = 'BEST_checkpoint.tar'
checkpoint = torch.load(checkpoint,map_location='cpu')
model = checkpoint['model'].module
model = model.to(device)
Thanks @madhukirand.
i've downloaded the model but both 7z and tar reports unknown format.
any suggestions what to check? i'm on windows, if it matters