Method
__init__
(self, use_h, use_w, rotate = 1, offset=False, ratio = 0.5, mode=0, prob = 1.)
Source from the content-addressed store, hash-verified
| 69 | |
| 70 | class GridMask(nn.Module): |
| 71 | def __init__(self, use_h, use_w, rotate = 1, offset=False, ratio = 0.5, mode=0, prob = 1.): |
| 72 | super(GridMask, self).__init__() |
| 73 | self.use_h = use_h |
| 74 | self.use_w = use_w |
| 75 | self.rotate = rotate |
| 76 | self.offset = offset |
| 77 | self.ratio = ratio |
| 78 | self.mode = mode |
| 79 | self.st_prob = prob |
| 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() |
Callers
nothing calls this directly
Tested by
no test coverage detected