core._cli

Copyright (c) Facebook, Inc. and its affiliates.

This source code is licensed under the MIT license found in the LICENSE file in the root directory of this source tree.

Classes

Runner

Derived callable classes are requeued after timeout with their current

Functions

runner_wrapper(config)
main([args, override_args])	Run the main fairchem program.

Module Contents

class core._cli.Runner

Bases: submitit.helpers.Checkpointable

Derived callable classes are requeued after timeout with their current state dumped at checkpoint.

__call__ method must be implemented to make your class a callable.

Note

The following implementation of the checkpoint method resubmits the full current state of the callable (self) with the initial argument. You may want to replace the method to curate the state (dump a neural network to a standard format and remove it from the state so that not to pickle it) and change/remove the initial parameters.

config = None

__call__(config: dict) → None

checkpoint(*args, **kwargs)

Resubmits the same callable with the same arguments

core._cli.runner_wrapper(config: dict)

core._cli.main(args: argparse.Namespace | None = None, override_args: list[str] | None = None)

Run the main fairchem program.