You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
17 lines
326 B
Python
17 lines
326 B
Python
8 months ago
|
import os
|
||
|
|
from abc import ABC, abstractmethod
|
||
|
|
|
||
|
|
|
||
|
|
class BaseLogger(ABC):
|
||
|
|
def __init__(self, save_dir):
|
||
|
|
self.save_dir = save_dir
|
||
|
|
os.makedirs(self.save_dir, exist_ok=True)
|
||
|
|
|
||
|
|
@abstractmethod
|
||
|
|
def log_metrics(self, metrics, prefix=None):
|
||
|
|
pass
|
||
|
|
|
||
|
|
@abstractmethod
|
||
|
|
def close(self):
|
||
|
|
pass
|