NMT-LaVi / models /default.py
hieungo1410's picture
'add'
8cb4f3b
raw
history blame
480 Bytes
class MockModel:
"""A model that only output string to show flow"""
def __init__(self, *args, **kwargs):
print("Mock model initialization, with args/kwargs: {} {}".format(args, kwargs))
def run_train(self, **kwargs):
print("Model in training, with args: {}".format(kwargs))
def run_eval(self, **kwargs):
print("Model in evaluation, with args: {}".format(kwargs))
def run_debug(self, **kwargs):
print("Model in debuging, with args: {}".format(kwargs))