(self, epoch, max_epoch)
| 80 | self.prob = prob |
| 81 | self.fp16_enable = False |
| 82 | def set_prob(self, epoch, max_epoch): |
| 83 | self.prob = self.st_prob * epoch / max_epoch #+ 1.#0.5 |
| 84 | @auto_fp16() |
| 85 | def forward(self, x): |
| 86 | if np.random.rand() > self.prob or not self.training: |
nothing calls this directly
no outgoing calls
no test coverage detected