(self, multiplier=1, shape=None, device=None)
| 226 | drop /= drop.mean() |
| 227 | weight *= drop |
| 228 | return weight |
| 229 | |
| 230 | def get_diff_weight(self, multiplier=1, shape=None, device=None): |
| 231 | scale = self.scale * multiplier |
| 232 | diff = self.get_weight(shape) * scale |
| 233 | if device is not None: |
| 234 | diff = diff.to(device) |
| 235 | return diff, None |
| 236 |
no test coverage detected