在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 *
2022-03-11 10:31:41
34KB
Python
1