文件名称：BYOL-Pytorch_lightning
文件大小：34KB
文件格式：ZIP
更新时间：2024-05-13 12:01:42
Python
在Pytorch中引导您自己的潜能（BYOL） 此代码存储库是基于byol-pytorch 0.5.2修改的。 主要更改包括： 添加了检查点模块。 添加了检查点转换模块（可以将保存的Pytorch Lightning检查点转换为Pytorch检查点模式。） 安装 $ pip install pytorch-lightning $ pip install pillow 用法 将单GPU或多GPU与Pytorch闪电配合使用： # Training $ python train . py - - image_folder / path / to / your / images # Ckpt convert $ python ckpt_convert . py - - ckpt_path * . ckpt - - save_path * . pth - - arch resnet *
【文件预览】：
BYOL-Pytorch_lightning-master
----ckpt_convert.py(2KB)
----train.py(3KB)
----BYOL.png(28KB)
----README.md(2KB)
----byol_pytorch()
--------byol_pytorch.py(8KB)
--------__init__.py(43B)