Hopenet
Note
import pathlib
import torch
paths = sorted(pathlib.Path('orig').glob('*'))
out_dir = pathlib.Path('models')
out_dir.mkdir(exist_ok=True)
for path in paths:
ckpt = torch.load(path, map_location='cpu')
torch.save(ckpt, out_dir / path.name)