(self)
| 56 | CLS = GradSampleModule |
| 57 | |
| 58 | def setUp(self): |
| 59 | self.original_model = SampleConvNet() |
| 60 | copy_of_original_model = SampleConvNet() |
| 61 | copy_of_original_model.load_state_dict( |
| 62 | self.original_model.state_dict(), strict=True |
| 63 | ) |
| 64 | |
| 65 | self.grad_sample_module = self.CLS( |
| 66 | copy_of_original_model, batch_first=True, loss_reduction="mean" |
| 67 | ) |
| 68 | self.DATA_SIZE = 8 |
| 69 | self.setUp_data() |
| 70 | self.criterion = nn.L1Loss() |
| 71 | |
| 72 | def setUp_data(self): |
| 73 | self.ds = FakeData( |
nothing calls this directly
no test coverage detected