↓ 2 callersMethod__init__(self, block, layers, num_classes=1000, with_neck=False, pretrain_path=None, enable_fc=False)
Bamboo-Benchmark/models/resnet.py:98
Method__init__(self, width: int, layers: int, heads: int, attn_mask: torch.Tensor = None, checkpoint: bool = False)
Bamboo-Benchmark/models/text_transformer.py:56