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