@file: main.py
@time: 2020/9/28 16:31
@desc:
'''
import os
import datetime
from DataSet import Dataset
from utils import *
import tensorflow as tf
from dssm import DSSM
config = Config()
dataset = Dataset()
nwords = dataset._vocab_size
trainData, evalData = dataset.dataGen()
train_epoch_steps = int(len(trainData) / Config.batchSize) - 1
eval_epoch_steps = int(len(evalData) / Config.batchSize) - 1
# 定义计算图
with tf.Graph().as_default():
session_conf = tf.ConfigProto(allow_soft_placement=True,
log_device_placement=False,
device_count={"CPU": 78})
sess = tf.Session(config=session_conf)
# 定义会话
with sess.as_default():
dssm = DSSM(config, nwords)
