(self)
| 153 | |
| 154 | @torch.jit.ignore |
| 155 | def no_weight_decay_keywords(self): |
| 156 | if hasattr(self.encoder, 'no_weight_decay_keywords'): |
| 157 | return {'encoder.' + i for i in self.encoder.no_weight_decay_keywords()} |
| 158 | return {} |
| 159 | |
| 160 | |
| 161 | def build_simmim(config): |