Multi-task classification project, with the custom training loop implemented from scratch in TF 2.2+ and the usage of "tf.data.Dataset" object for handling the data.
template computer-vision tensorflow cv keras cnn tf2 mnist tf-data augmentation cnn-classification multitask-learning mlflow tf-function gradient-tape custom-training-loops
-
Updated
Sep 19, 2022 - Python