Add a flag for end of QAT (#297)

Signed-off-by: Cheng, Penghui <[email protected]>
Signed-off-by: zehao-intel <[email protected]>

Author: PenghuiCheng

neural_compressor/training.py

@@ -77,6 +77,7 @@ def on_train_begin(self, dataloader=None):
         def on_train_end(self):
             """ called after the end of epochs"""
             self.callbacks.on_train_end()
+            logger.info("Training finished!")
 
         def on_epoch_begin(self, epoch):
             """ called on the beginning of epochs"""