File size: 590 Bytes
b1882a9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | """Architecture selection including the projected scalar warmup experiment."""
import torch
from tinyasr.codec_factory_v1 import load_codec_checkpoint as load_previous
from tinyasr.codec_projected_warmup_v1 import ARCHITECTURE, ProjectedWarmupCodec
def load_codec_checkpoint(path):
checkpoint = torch.load(path, map_location='cpu', weights_only=False)
if checkpoint.get('architecture') != ARCHITECTURE:
return load_previous(path)
model = ProjectedWarmupCodec()
model.load_state_dict(checkpoint['codec'], strict=True)
return model.eval().requires_grad_(False)
|