Spaces:
Runtime error
Runtime error
File size: 597 Bytes
54660f7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from simclr import SimCLR
import yaml
from data_aug.dataset_wrapper import DataSetWrapper
import os, glob
import pandas as pd
import argparse
def main():
parser = argparse.ArgumentParser()
parser.add_argument('--magnification', type=str, default='20x')
parser.add_argument('--dest_weights', type=str)
args = parser.parse_args()
config = yaml.load(open("config.yaml", "r"), Loader=yaml.FullLoader)
dataset = DataSetWrapper(config['batch_size'], **config['dataset'])
simclr = SimCLR(dataset, config, args)
simclr.train()
if __name__ == "__main__":
main()
|