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

Don't eval after load

Browse source
This commit is contained in:
Andras Schmelczer 2024-09-01 22:10:31 +01:00
parent 9e286264b8
commit a5d74c650c
No known key found for this signature in database
GPG key ID: FC8F2C3D3D1A718C
3 changed files with 2245 additions and 2199 deletions
Download patch file Download diff file Expand all files Collapse all files

1
src/models/__init__.py
View file

@ -51,7 +51,6 @@ def load_model(path: Path, device: torch.device) -> Tuple[nn.Module, Dict[str, A
device=device,
)
model.load_state_dict(torch.load(model_path))
model.eval()
logging.info(f"Parameter count: {sum(p.numel() for p in model.parameters())}")
return model, hyperparameters