Skip to content

Latest commit

 

History

History
15 lines (11 loc) · 713 Bytes

README.md

File metadata and controls

15 lines (11 loc) · 713 Bytes

AutoTrainer

Setup

To install all dependencies, please run:

$ pip -r requirements.txt

Usage

The main function of AUTOTRAINER is model_train().This function accepts parameters like model, training configuration set as input. For details, please refer to the corresponding function comments.
There are two ways to have a easy start on AUTOTRAINER:

  1. You can just modify reproduce.py to load your own dataset and then use your training configuration file and model.
  2. You can also call model_train() to use AUTOTRAINER. In this method, you need to make sure your codes call this function from the correct file and pass the necessary data to it.