From cf82361a054776c850e61e76511b39f264411211 Mon Sep 17 00:00:00 2001 From: tangwei Date: Fri, 10 Apr 2020 13:40:56 +0800 Subject: [PATCH] code fix --- fleetrec/models/base.py | 2 +- setup.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/fleetrec/models/base.py b/fleetrec/models/base.py index 22cedc06..763c3666 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 d5de390d..39a1b7ab 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( -- GitLab