fix bugs(update src)

This commit is contained in:
wangshuai6
2025-04-14 08:45:47 +08:00
parent ea60d28838
commit 9010659f72
2 changed files with 1 additions and 2 deletions

View File

@@ -3,7 +3,6 @@ from typing import Optional, Dict, Any
import lightning.pytorch as pl import lightning.pytorch as pl
from lightning.pytorch.callbacks.model_checkpoint import ModelCheckpoint from lightning.pytorch.callbacks.model_checkpoint import ModelCheckpoint
from soupsieve.util import lower
class CheckpointHook(ModelCheckpoint): class CheckpointHook(ModelCheckpoint):

View File

@@ -15,7 +15,7 @@ def process_fn(image, path):
Image.fromarray(image).save(path) Image.fromarray(image).save(path)
class SaveImagesHook(Callback): 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.save_dir = save_dir
self.max_save_num = max_save_num self.max_save_num = max_save_num
self.compressed = compressed self.compressed = compressed