update model

src/main.py

@@ -21,7 +21,7 @@ from models.esmm.model import esmm_model_fn, model_export, model_predict_diary
 def main():
     time_begin = time.time()
 
-    tf.logging.set_verbosity(tf.compat.v1.logging.INFO)
+    tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.INFO)
 
     # os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 

src/main_portrait.py

@@ -34,7 +34,7 @@ def user_portrait_scan_info():
 
 def main():
     time_begin = time.time()
-    tf.logging.set_verbosity(tf.compat.v1.logging.INFO)
+    tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.INFO)
 
     device_dict = get_device_dict_from_redis()
     diary_dict = get_diary_dict_from_redis()
@@ -50,7 +50,7 @@ def main():
     # user_portrait_scan_info()
 
     total_time = (time.time() - time_begin) / 60
-    print("total cost {:.2f} mins at {}".format(total_time, datetime.now()))
+    print("total cost {:.2f} mins at {}".format(total_time, datetime.datetime.now()))
 
 
 if __name__ == "__main__":