improve plugins
This commit is contained in:
parent
42c417171d
commit
f83934f5d3
|
@ -0,0 +1,19 @@
|
||||||
|
from plugins.plugins import BasePlugin
|
||||||
|
import logging
|
||||||
|
from colorama import Fore, Style
|
||||||
|
|
||||||
|
class ExampleLoggingPlugin(BasePlugin):
|
||||||
|
def __init__(self):
|
||||||
|
print(f"{Fore.LIGHTBLUE_EX}ExampleLoggingPlugin init{Style.RESET_ALL}")
|
||||||
|
# Setup any state variables here
|
||||||
|
pass
|
||||||
|
|
||||||
|
def on_epoch_start(self, **kwargs):
|
||||||
|
logging.info(f"{Fore.LIGHTBLUE_EX} ** ExampleLoggingPlugin: on_epoch_start{Style.RESET_ALL}")
|
||||||
|
for k, v in kwargs.items():
|
||||||
|
logging.info(f" {Fore.BLUE}{k}: {v}{Style.RESET_ALL}")
|
||||||
|
|
||||||
|
def on_epoch_end(self, **kwargs):
|
||||||
|
logging.info(f"{Fore.LIGHTBLUE_EX} ** ExampleLoggingPlugin: on_epoch_end{Style.RESET_ALL}")
|
||||||
|
for k, v in kwargs.items():
|
||||||
|
logging.info(f" {Fore.BLUE}{k}: {v}{Style.RESET_ALL}")
|
Loading…
Reference in New Issue