pytorch_igniter_demo package¶
Submodules¶
pytorch_igniter_demo.config module¶
-
class
pytorch_igniter_demo.config.Trainer(args, model)¶ Bases:
torch.nn.modules.module.Module-
step(engine, batch)¶
-
-
pytorch_igniter_demo.config.eval_args(parser)¶
-
pytorch_igniter_demo.config.make_config()¶
-
pytorch_igniter_demo.config.model_args(parser)¶
-
pytorch_igniter_demo.config.train_args(parser)¶
pytorch_igniter_demo.dataprep module¶
-
class
pytorch_igniter_demo.dataprep.DataprepCommand(env=None)¶ Bases:
aws_sagemaker_remote.processing.main.ProcessingCommand
-
pytorch_igniter_demo.dataprep.dataprep(args)¶
-
pytorch_igniter_demo.dataprep.get_dataset(path, train=True)¶
-
pytorch_igniter_demo.dataprep.get_loader(path, batch_size, train=True, shuffle=True)¶
pytorch_igniter_demo.main module¶
-
pytorch_igniter_demo.main.main(dry_run=False)¶ Generate experiment CLI. Running locally, this function is run by a wrapper created by setuptools
-
pytorch_igniter_demo.main.parser_for_docs()¶ Create a parser for generating documentation