File size: 188 Bytes
a3f6fc6 | 1 2 3 4 5 6 7 | import torch
# for supporting output_dim as metadata for the config
def from_ckpt(f: str, output_dim: int, n_patches: int):
model = torch.load(f, weights_only=False)
return model |
a3f6fc6 | 1 2 3 4 5 6 7 | import torch
# for supporting output_dim as metadata for the config
def from_ckpt(f: str, output_dim: int, n_patches: int):
model = torch.load(f, weights_only=False)
return model |