| 331 | small_kernel_merged=small_kernel_merged) |
| 332 | |
| 333 | def create_RepLKNetXL(drop_path_rate=0.3, num_classes=1000, use_checkpoint=True, small_kernel_merged=False): |
| 334 | return RepLKNet(large_kernel_sizes=[27,27,27,13], layers=[2,2,18,2], channels=[256,512,1024,2048], |
| 335 | drop_path_rate=drop_path_rate, small_kernel=None, dw_ratio=1.5, |
| 336 | num_classes=num_classes, use_checkpoint=use_checkpoint, |
| 337 | small_kernel_merged=small_kernel_merged) |
| 338 | |
| 339 | if __name__ == '__main__': |
| 340 | model = create_RepLKNet31B(small_kernel_merged=False) |