Pruning a Neural Network trained on a balanced dataset
Start with main.py
#tensor([0.4942, 0.4851, 0.4504]) tensor([0.2466, 0.2429, 0.2616]) --> testset #tensor([0.4914, 0.4822, 0.4465]) tensor([0.2470, 0.2435, 0.2616]) --> trainset balanced #tensor([0.4904, 0.4792, 0.4396]) tensor([0.2465, 0.2428, 0.2597]) --> trainset imbalanced