fix bugs(update src)
This commit is contained in:
@@ -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):
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user