Hopenet / README.md
hysts's picture
hysts HF Staff
Add files
373e0bb

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)