BYOL-Pytorch_lightning

在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 *