(pretrain_path)
| 921 | |
| 922 | |
| 923 | def at_base(pretrain_path): |
| 924 | student = vit_base() |
| 925 | weight = torch.load(pretrain_path, map_location='cpu') |
| 926 | msg = student.load_state_dict(weight, strict=False) |
| 927 | print(msg) |
| 928 | model = AttributeTransformer(student) |
| 929 | return model |
| 930 | |
| 931 | |
| 932 | def at2_base(pretrain_path, num_attribute=28, feat_channal=197, grad_from_block=11): |
nothing calls this directly
no test coverage detected