Create Node2Vec ModuleKeeper

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/18504

Reviewed By: sunnieshang

Differential Revision: D14632091

fbshipit-source-id: d4544866552dc6bcbc7515be9e88cb11e7622a44

diff --git a/caffe2/operators/text_file_reader.cc b/caffe2/operators/text_file_reader.cc
index 4afd689..cbe3d81 100644 (file)
--- a/caffe2/operators/text_file_reader.cc
+++ b/caffe2/operators/text_file_reader.cc
@@ -169,6 +169,7 @@ REGISTER_CPU_OPERATOR(TextFileReaderRead, TextFileReaderReadOp);
 OPERATOR_SCHEMA(CreateTextFileReader)
     .NumInputs(0)
     .NumOutputs(1)
+    .ScalarType(TensorProto::UNDEFINED)
     .SetDoc("Create a text file reader. Fields are delimited by <TAB>.")
     .Arg("filename", "Path to the file.")
     .Arg("num_passes", "Number of passes over the file.")

diff --git a/caffe2/python/text_file_reader.py b/caffe2/python/text_file_reader.py
index c351bcd..52a1b27 100644 (file)
--- a/caffe2/python/text_file_reader.py
+++ b/caffe2/python/text_file_reader.py
@@ -21,7 +21,7 @@ class TextFileReader(Reader):
             init_net   : Net that will be run only once at startup.
             filename   : Path to file to read from.
             schema     : schema.Struct representing the schema of the data.
-                         Currently, only support Struct of strings.
+                         Currently, only support Struct of strings and float32.
             num_passes : Number of passes over the data.
             batch_size : Number of rows to read at a time.
         """