mirror of
https://github.com/sthalles/SimCLR.git
synced 2025-06-03 15:03:00 +08:00
14 lines
213 B
Python
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()
|