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.
templatecomputer-visiontensorflowcvkerascnntf2mnisttf-dataaugmentationcnn-classificationmultitask-learningmlflowtf-functiongradient-tapecustom-training-loops
-
Updated
Sep 19, 2022 - Python