oracle.train

Interface for training models in the ORACLE framework.

Functions

get_wandb_run(args)

Initializes and returns a Weights & Biases (wandb) run with the specified configuration.

main()

parse_args()

Get commandline options

run_training_loop(args)

Runs the training loop for the model using the specified configuration and dataset loaders.

save_args_to_csv(args, filepath)

Save command-line arguments to a CSV file.