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)