| 376 | |
| 377 | |
| 378 | class Tiger_Transformer_custom(Tiger_Transformer): |
| 379 | sa_blocks = [ |
| 380 | ((32, 2, 32), (1024, 0.1, 32, (32, 64))), |
| 381 | ((64, 3, 16), (256, 0.2, 32, (64, 128))), |
| 382 | ((128, 3, 8), (64, 0.4, 32, (128, 256))), |
| 383 | (None, (16, 0.8, 32, (256, 256, 512))), |
| 384 | ] |
| 385 | fp_blocks = [ |
| 386 | ((256, 256), (256, 3, 8)), |
| 387 | ((256, 256), (256, 3, 8)), |
| 388 | ((256, 128), (128, 2, 16)), |
| 389 | ((128, 128, 64), (64, 2, 32)), |
| 390 | ] |
| 391 | |
| 392 | def __init__(self, num_classes, embed_dim, use_att,dropout, extra_feature_channels=3, width_multiplier=1, |
| 393 | voxel_resolution_multiplier=1): |
| 394 | super().__init__( |
| 395 | num_classes=num_classes, embed_dim=embed_dim, use_att=use_att, |
| 396 | dropout=dropout, extra_feature_channels=extra_feature_channels, |
| 397 | width_multiplier=width_multiplier, voxel_resolution_multiplier=voxel_resolution_multiplier |
| 398 | ) |
| 399 | |
| 400 | |
| 401 | |