SimCLR/run.py
2020-03-13 22:56:04 -03:00

14 lines
213 B
Python

from simclr import SimCLR
import yaml
def main():
config = yaml.load(open("config.yaml", "r"), Loader=yaml.FullLoader)
simclr = SimCLR(config)
simclr.train()
if __name__ == "__main__":
main()