UniTrain
UniTrain copied to clipboard
Classification: Add a support for LeNet-5 for the classification model