Exemplos de TrialMNISTDataModule.add_argparse_args em Python

Exemplo n.º 1

Arquivo: test_datamodules.py Projeto: uiuc-arc/pytorch-lightning

def test_dm_init_from_argparse_args(tmpdir):
    parser = ArgumentParser()
    parser = TrialMNISTDataModule.add_argparse_args(parser)
    args = parser.parse_args(['--data_dir', './my_data'])
    dm = TrialMNISTDataModule.from_argparse_args(args)
    dm.prepare_data()
    dm.setup()

Exemplo n.º 2

Arquivo: test_datamodules.py Projeto: uiuc-arc/pytorch-lightning

def test_dm_add_argparse_args(tmpdir):
    parser = ArgumentParser()
    parser = TrialMNISTDataModule.add_argparse_args(parser)
    args = parser.parse_args(['--data_dir', './my_data'])
    assert args.data_dir == './my_data'