Views
No views yet
1import pathlib
2
3import torch
4
5paths = sorted(pathlib.Path('orig').glob('*'))
6
7out_dir = pathlib.Path('models')
8out_dir.mkdir(exist_ok=True)
9
10for path in paths:
11 ckpt = torch.load(path, map_location='cpu')
12 torch.save(ckpt, out_dir / path.name)