From 00f46a574dd145abae4912d23073dccf37dbc9df Mon Sep 17 00:00:00 2001 From: chengmo Date: Tue, 31 Mar 2020 17:26:04 +0800 Subject: [PATCH] fix --- PaddleRec/tdm/tdm_demo/train_network.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PaddleRec/tdm/tdm_demo/train_network.py b/PaddleRec/tdm/tdm_demo/train_network.py index ac32d88f..cca86d20 100644 --- a/PaddleRec/tdm/tdm_demo/train_network.py +++ b/PaddleRec/tdm/tdm_demo/train_network.py @@ -96,7 +96,7 @@ class TdmTrainNet(object): output_positive=self.output_positive, output_list=True, seed=0, - dtype='int32' + dtype='int64' ) trace_var(sample_nodes, "[TDM][tdm_sample]", -- GitLab