【文件属性】：
文件名称：BYOL-Pytorch_lightning
文件大小：34KB
文件格式：ZIP
更新时间：2021-04-23 18:15:02
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)