Bug fix, improvements

commanderxa · Oct 8, 2023 · b9c52eb · b9c52eb
1 parent 02268ae
commit b9c52eb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/nn/dscovry/dataset.py b/nn/dscovry/dataset.py
@@ -53,7 +53,6 @@ def __init__(self, annotation_files: list[str]) -> None:
  )
  )
  self.data = pd.concat(data, axis=0, ignore_index=True)
- print(len(self.data))
 
  def __len__(self) -> int:
  return len(self.data)

diff --git a/nn/main.py b/nn/main.py
@@ -24,7 +24,7 @@ def my_app(cfg: DictConfig) -> None:
  # model = torch.compile(model)
 
  # load the model
- checkpoint = torch.load(f"models/{cfg.model.name}_{cfg.hyper.n_hidden}_a.pt")
+ checkpoint = torch.load(f"models/{cfg.model.name}_{cfg.hyper.n_hidden}.pt")
  model.load_state_dict(checkpoint["model"])
  model.eval()
 
@@ -69,8 +69,9 @@ def evaluate(model: DSCOVRYModel, dataloader: DataLoader, cfg):
 
  accuracy = epoch_accuracy / len(dataloader)
  loss = epoch_loss / len(dataloader)
- print(f"Accuracy ({Config.cfg.hyper.tolerance} tolearnce): {accuracy}%")
- print(f"Loss: {loss}")
+ print(f"\n|\n| Accuracy: {accuracy:.2f}%")
+ print(f"| Accuracy tolerance: {Config.cfg.hyper.tolerance}")
+ print(f"| Loss: {loss}\n|\n")
 
 
 if __name__ == "__main__":