train.py

The main function for training LOIS.

python train.py --problem=$problem_name --save_path=$path_to_the_saved_model

• Arugments:
  • problem: The name of the training problem.
  • save_path: The path to the saved model.