def make_model():
global model
net_arch = args # en1_units, en2_units, num_topic, num_input
net_arch.num_input = vocab_size
model = ProdLDA(net_arch)
if not args.nogpu:
model = model.cuda()
def make_model():
global model
net_arch = args # en1_units, en2_units, num_topic, num_input
net_arch.num_input = data_tr.shape[1]
model = ProdLDA(net_arch)
if not args.nogpu:
model = model.cuda()