Exemplos de distributed_cmd_run em Python

Exemplo n.º 1

Arquivo: run.py Projeto: snehashischatterjee1997/catalyst

def main(args, unknown_args):
    """Run the ``catalyst-dl run`` script."""
    distributed_cmd_run(main_worker, args.distributed, args, unknown_args)

Exemplo n.º 2

def main():
    utils.distributed_cmd_run(train)

Exemplo n.º 3

Arquivo: z_distributed_17.py Projeto: ngxbac/catalyst

# 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)