PyTorch-BYOL 潜能的PyTorch实施。 安装 克隆存储库并运行 $ conda env create --name byol --file env.yml $ conda activate byol $ python main.py 设定档 在运行PyTorch BYOL之前,请确保在config.yaml文件上选择了正确的运行配置。 network : name : resnet18 # base encoder. choose one of resnet18 or resnet50 # Specify a folder containing a pre-trained model to fine-tune. If training from scratch, pass None. fine_tune_from : ' resnet-18_40-epo