- Original code from https://github.com/sthalles/PyTorch-BYOL
- Setup
config.yaml
: BYOL related configuration. - Setup
tspipe.yaml
: TSPipe related configuration.
python main.py --tspipe-config=tspipe.yaml --ip=localhost --rank=0 --num-nodes=1
Name | Name | Last commit date | ||
---|---|---|---|---|
parent directory.. | ||||
config.yaml
: BYOL related configuration.tspipe.yaml
: TSPipe related configuration.python main.py --tspipe-config=tspipe.yaml --ip=localhost --rank=0 --num-nodes=1