andras/bipolaroidbipolaroid
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix errors

Browse source
This commit is contained in:
Andras Schmelczer 2024-06-27 23:13:37 +01:00
parent 7863611f86
commit 35eb747abf
No known key found for this signature in database
GPG key ID: FC8F2C3D3D1A718C
3 changed files with 26 additions and 73 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/models/__init__.py
View file

@ -47,7 +47,7 @@ def load_model(path: Path, device: torch.device) -> Tuple[nn.Module, Dict[str, A
model_path = path.with_suffix(".pth")
model = create_model(
type=hyperparameters["model_type"],
**hyperparameters,
hyperparameters=hyperparameters,
device=device,
)
model.load_state_dict(torch.load(model_path))