b940bd9174414ecef9ac99736ce1b2acaf7362d8,torchsummary/tests/unit_tests/torchsummary_test.py,,,#,46

Before Change

        self.assertEqual(total_params, 31120)
        self.assertEqual(trainable_params, 31120)

if __name__ == "__main__":
    unittest.main(buffer=True)

After Change

from torchsummary.tests.test_models.test_model import SingleInputNet, MultipleInputNet, MultipleInputNetDifferentDtypes
import torch

gpu_if_available = "cuda:0" if torch.cuda.is_available() else "cpu"

class torchsummaryTests(unittest.TestCase):
    def test_single_input(self):
        model = SingleInputNet()

Instances