Refactor
Refactoring the code for custom data training.
-
Upgrade to TF 2.x -
Add .pre-commithook -
Support CPU/GPU switching -
Add data pipeline for custom data -
Prepare a small portion of data for training
Refactoring the code for custom data training.
.pre-commit hook