Merge branch 'tdm_dev' of https://github.com/MrChengmo/models into tdm_dev

7fb1af29 · chengmo · a503d585 · 00f46a57 · 7fb1af29
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

PaddleRec/tdm/tdm_demo/train_network.py PaddleRec/tdm/tdm_demo/train_network.py +1 -1

未找到文件。
--- 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]",