extensive-vision-ai / TheTensorClan

MIT License
0 stars 1 forks source link

create different trainers for different tasks #8

Open satyajitghana opened 4 years ago

satyajitghana commented 4 years ago

Now that we have to train GANs as well, why not have different Trainers for different tasks, like

ClassifierTrainer
GANTrainer
LSTMTrainer
BERTTrainer

each of the trainer is responsible to know how to train that kind of network and has the training loop and test loop in them, we can expose an api for users to add their own Trainers !