test/fastNLP
1
0
Fork 0
mirror of https://gitee.com/fastnlp/fastNLP.git synced 2024-12-02 04:07:35 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge branch 'dev' of github.com:choosewhatulike/fastNLP-private into dev

Browse Source
This commit is contained in:
yh 2019-01-11 19:55:54 +08:00
commit a1fe24c49d
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
fastNLP/core/callback.py
View File

@ -177,34 +177,36 @@ class DummyCallback(Callback):
def before_train(self, *arg):
print(arg)
def after_epoch(self):
print("after epoch!!!")
return 12
def after_epoch(self, cur_epoch, n_epoch, optimizer):
print(cur_epoch, n_epoch, optimizer)
class EchoCallback(Callback):
def before_train(self):
print("before_train")
def before_epoch(self):
def before_epoch(self, cur_epoch, total_epoch):
print("before_epoch")
def before_batch(self):
def before_batch(self, batch_x, batch_y, indices):
print("before_batch")
print("batch_x:", batch_x)
print("batch_y:", batch_y)
print("indices: ", indices)
def before_loss(self):
def before_loss(self, batch_y, predict_y):
print("before_loss")
def before_backward(self):
def before_backward(self, loss, model):
print("before_backward")
def after_batch(self):
print("after_batch")
def after_epoch(self):
def after_epoch(self, cur_epoch, n_epoch, optimizer):
print("after_epoch")
def after_train(self):
def after_train(self, model):
print("after_train")
class GradientClipCallback(Callback):