Go back to training mode after evaluation

nilozdemir · Jul 29, 2022 · 810d280 · 810d280
1 parent 063ac98
commit 810d280
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 4 deletions.
diff --git a/CITATION b/CITATION
@@ -13,13 +13,11 @@
 
 @incollection{BARS,
   author    = {Jieming Zhu and
-               Kelong Mao and
                Quanyu Dai and
                Liangcai Su and
                Rong Ma and
                Jinyang Liu and
                Guohao Cai and
-               Zhicheng Dou and
                Xi Xiao and
                Rui Zhang},
   title     = {BARS: Towards Open Benchmarking for Recommender Systems},

diff --git a/README.md b/README.md
@@ -27,6 +27,8 @@ This repo is the community dev version of the original release at [huawei-noah/b
 
 > Jieming Zhu, Jinyang Liu, Shuai Yang, Qi Zhang, Xiuqiang He. [Open Benchmarking for Click-Through Rate Prediction](https://arxiv.org/abs/2009.05794). *The 30th ACM International Conference on Information and Knowledge Management (CIKM)*, 2021. [[Bibtex](https://dblp.org/rec/conf/cikm/ZhuLYZH21.html?view=bibtex)]
 
+> Jieming Zhu, Quanyu Dai, Liangcai Su, Rong Ma, Jinyang Liu, Guohao Cai, Xi Xiao, Rui Zhang. [BARS: Towards Open Benchmarking for Recommender Systems](https://arxiv.org/abs/2205.09626). *The 45th International ACM SIGIR Conference on Research and Development in Information Retrieval (SIGIR)*, 2022. [Bibtex]
+
 ## Key Features
 
 + **Configurable**: Both data preprocessing and models are modularized and configurable.

diff --git a/fuxictr/pytorch/models/base_model.py b/fuxictr/pytorch/models/base_model.py
@@ -132,6 +132,7 @@ def on_batch_end(self, batch, logs={}):
             epoch = round(float(self._total_batches) / self._batches_per_epoch, 2)
             val_logs = self.evaluate_generator(self.valid_gen)
             self.checkpoint_and_earlystop(epoch, val_logs)
+            self.train()
             logging.info("--- {}/{} batches finished ---".format(batch + 1, self._batches_per_epoch))
 
     def lr_decay(self, factor=0.1, min_lr=1e-6):

diff --git a/fuxictr/version.py b/fuxictr/version.py
@@ -1 +1 @@
-__version__ = "1.2.2"
+__version__ = "1.2.3"
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="fuxictr",
-    version="1.2.2",
+    version="1.2.3",
     author="zhujiem",
     author_email="[email protected]",
     description="A configurable, tunable, and reproducible library for CTR prediction",