Closed Naireen closed 4 years ago
Addressing issue 90, where there are issues with calculating the correct number of parameters . Also added python unit testing testing single and multiple inputs. Added return values to the torch summary function, as it allows for cleaner testing.
Addressing issue 90, where there are issues with calculating the correct number of parameters . Also added python unit testing testing single and multiple inputs. Added return values to the torch summary function, as it allows for cleaner testing.