From 9010659f723685878c4e7b453a7c90d068680b9f Mon Sep 17 00:00:00 2001 From: wangshuai6 Date: Mon, 14 Apr 2025 08:45:47 +0800 Subject: [PATCH] fix bugs(update src) --- src/callbacks/model_checkpoint.py | 1 - src/callbacks/save_images.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/src/callbacks/model_checkpoint.py b/src/callbacks/model_checkpoint.py index 1e9b8e6..7a5df2c 100644 --- a/src/callbacks/model_checkpoint.py +++ b/src/callbacks/model_checkpoint.py @@ -3,7 +3,6 @@ from typing import Optional, Dict, Any import lightning.pytorch as pl from lightning.pytorch.callbacks.model_checkpoint import ModelCheckpoint -from soupsieve.util import lower class CheckpointHook(ModelCheckpoint): diff --git a/src/callbacks/save_images.py b/src/callbacks/save_images.py index c6cd32b..85e9c0b 100644 --- a/src/callbacks/save_images.py +++ b/src/callbacks/save_images.py @@ -15,7 +15,7 @@ def process_fn(image, path): Image.fromarray(image).save(path) class SaveImagesHook(Callback): - def __init__(self, save_dir="val", max_save_num=0, compressed=True): + def __init__(self, save_dir="val", max_save_num=100, compressed=True): self.save_dir = save_dir self.max_save_num = max_save_num self.compressed = compressed