MAIN

This script is the main entry point for the Mammoth project. It contains the main function main() that orchestrates the training process.

The script performs the following tasks: - Imports necessary modules and libraries. - Sets up the necessary paths and configurations. - Parses command-line arguments. - Initializes the dataset, model, and other components. - Trains the model using the train() function.

To run the script, execute it directly or import it as a module and call the main() function.

Functions

utils.main.check_args(args, dataset=None)

Just a (non complete) stream of asserts to ensure the validity of the arguments.

utils.main.extend_args(args, dataset)

Extend the command-line arguments with the default values from the dataset and the model.

utils.main.lecun_fix()

utils.main.load_configs(parser)

Return type:

dict

utils.main.main(args=None)

utils.main.parse_args()

Parse command line arguments for the mammoth program and sets up the args object.

Returns:

Parsed command line arguments.

Return type:

args (argparse.Namespace)