diff --git a/fleetrec/models/base.py b/fleetrec/models/base.py index 22cedc067c45a192da58e4796bcba5b4aab0d3c8..763c36663ea03eb21c16823e575bd3b6474a1256 100644 --- a/fleetrec/models/base.py +++ b/fleetrec/models/base.py @@ -78,7 +78,7 @@ class Model(object): self._cost = None self._metrics = {} self._data_var = [] - self._fetch_interval = 10 + self._fetch_interval = 20 def get_cost_op(self): """R diff --git a/setup.py b/setup.py index d5de390dbe7db96c6852a678f8c7d1747d9f27cb..39a1b7ab8158a6c97e4bd41957f3da43d61f6849 100644 --- a/setup.py +++ b/setup.py @@ -4,7 +4,8 @@ setup for fleet-rec. from setuptools import setup -packages = ["fleetrec", "fleetrec.examples", "fleetrec.metrics", "fleetrec.models", "fleetrec.reader", +packages = ["fleetrec", "fleetrec.examples", "fleetrec.metrics", "fleetrec.models", "fleetrec.models.ctr_dnn", + "fleetrec.reader", "fleetrec.trainer", "fleetrec.utils"] requires = [ @@ -19,7 +20,6 @@ about["__author__"] = "seiriosPlus" about["__author_email__"] = "tangwei12@baidu.com" about["__url__"] = "https://github.com/seiriosPlus/FleetRec" - readme = "..." setup(