diff --git a/src/autoencodix/base/_base_visualizer.py b/src/autoencodix/base/_base_visualizer.py index d5a71eec..c7db97c9 100644 --- a/src/autoencodix/base/_base_visualizer.py +++ b/src/autoencodix/base/_base_visualizer.py @@ -234,8 +234,6 @@ def _make_loss_format(result: Result, config: DefaultConfig) -> pd.DataFrame: loss_df = pd.DataFrame.from_dict(loss_values, orient="index") # type: ignore # Rest of your code remains the same - if term == "var_loss": - loss_df = loss_df * config.beta loss_df["Epoch"] = loss_df.index + 1 loss_df["Loss Term"] = term