def main(args, unknown_args): """Run the ``catalyst-dl run`` script.""" distributed_cmd_run(main_worker, args.distributed, args, unknown_args)
def main(): utils.distributed_cmd_run(train)
# and utils.distributed_cmd_run def train(): """ Training loop function. """ num_features = 10 model = Projector(num_features) runner = dl.SupervisedRunner() runner.train( model=model, # loaders={"train": loader, "valid": loader}, datasets={ "batch_size": 32, "num_workers": 1, "get_datasets_fn": datasets_fn, "num_features": num_features, }, criterion=nn.MSELoss(), optimizer=optim.Adam(model.parameters()), logdir="logs/log_example_17", num_epochs=10, verbose=True, check=True, fp16=True, distributed=True, # we should not get an error here, only warning ) utils.distributed_cmd_run(train)