simclr-converter:用于SimCLR检查点的PyTorch转换器

时间:2024-06-14 12:19:51
【文件属性】:

文件名称:simclr-converter:用于SimCLR检查点的PyTorch转换器

文件大小:10KB

文件格式:ZIP

更新时间:2024-06-14 12:19:51

Python

用于SimCLR检查点的PyTorch转换器 这是一个将库中提供的TensorFlow检查点转换为PyTorch格式的转换器,以方便相关研究。 用法 首先,从库下载并解压缩检查点,您将获得3个文件夹: ResNet50_1x , ResNet50_2x和ResNet50_4x 。 运行以下命令来转换3个检查点: python convert.py ResNet50_1x/model.ckpt-225206 resnet50-1x.pth python convert.py ResNet50_2x/model.ckpt-225206 resnet50-2x.pth python convert.py ResNet50_4x/model.ckpt-225206 resnet50-4x.pth 您将获得3个PyTorch检查点, resnet50-1x.pth , resnet50-2x.


【文件预览】:
simclr-converter-master
----convert.py(3KB)
----eval.py(5KB)
----LICENSE(11KB)
----README.md(2KB)
----resnet_wider.py(7KB)

网友评论