Overwrite model.train with this function to make sure train/eval mode does not change anymore.
(self, mode=True)
| 41 | |
| 42 | |
| 43 | def disabled_train(self, mode=True): |
| 44 | """Overwrite model.train with this function to make sure train/eval mode |
| 45 | does not change anymore.""" |
| 46 | return self |
| 47 | |
| 48 | |
| 49 | def uniform_on_device(r1, r2, shape, device): |
nothing calls this directly
no outgoing calls
no test coverage detected