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 *