File size: 163 Bytes
0b32ad6 |
1 2 3 4 5 6 7 8 9 |
import torch
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("path")
args = parser.parse_args()
torch.load(args.path, map_location="cpu")
|
0b32ad6 |
1 2 3 4 5 6 7 8 9 |
import torch
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("path")
args = parser.parse_args()
torch.load(args.path, map_location="cpu")
|