/pytorch_resnet50

I am using ResNet-50 to classify data with dimensions (37, 128, 1), where the labels have a shape of (batch_size, classes).

Primary LanguagePython

Stargazers