(pretrain_path, dict_attribute, grad_from_block=11)
| 1180 | |
| 1181 | |
| 1182 | def meta2_base(pretrain_path, dict_attribute, grad_from_block=11): |
| 1183 | student = vit_base() |
| 1184 | weight = torch.load(pretrain_path, map_location='cpu') |
| 1185 | msg = student.load_state_dict(weight, strict=False) |
| 1186 | print(msg) |
| 1187 | model = Meta_Attribute_Generator2(student, dict_attribute, grad_from_block=grad_from_block) |
| 1188 | return model |
nothing calls this directly
no test coverage detected