UniTrain
UniTrain copied to clipboard
Classification: Add a support for AlexNet for the classification model