From 74f062a0c543555ae0298221c56fcdf87c325257 Mon Sep 17 00:00:00 2001 From: lucidrains Date: Tue, 28 May 2024 09:36:02 -0700 Subject: [PATCH] oops --- alphafold3_pytorch/trainer.py | 2 +- pyproject.toml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/alphafold3_pytorch/trainer.py b/alphafold3_pytorch/trainer.py index 4f5e3607..fd1fe296 100644 --- a/alphafold3_pytorch/trainer.py +++ b/alphafold3_pytorch/trainer.py @@ -265,7 +265,7 @@ def __call__( scaled_valid_loss = valid_loss.item() * scale total_valid_loss += scaled_valid_loss - self.print(f'valid loss: {valid_loss.item():.3f}') + self.print(f'valid loss: {total_valid_loss:.3f}') self.wait() diff --git a/pyproject.toml b/pyproject.toml index 56cb6ad2..b11a3e9d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "alphafold3-pytorch" -version = "0.0.47" +version = "0.0.48" description = "Alphafold 3 - Pytorch" authors = [ { name = "Phil Wang", email = "lucidrains@gmail.com" }