(self, num_classes)
| 145 | self.is_remix = is_remix |
| 146 | |
| 147 | def build(self, num_classes): |
| 148 | return WideResNetVar( |
| 149 | first_stride=self.first_stride, |
| 150 | depth=self.depth, |
| 151 | num_classes=num_classes, |
| 152 | widen_factor=self.widen_factor, |
| 153 | drop_rate=self.dropRate, |
| 154 | is_remix=self.is_remix, |
| 155 | ) |
| 156 | |
| 157 | |
| 158 | if __name__ == '__main__': |