diff options
| author | Ru Li <ruxuanli@fb.com> | 2019-04-01 10:31:08 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-04-01 10:36:23 -0700 |
| commit | 3749d65a7ef9a327a165556ce2723e43da17c374 (patch) | |
| tree | ba73898071cf1ba8bce3cfbd66c3b3908f7a587e /caffe2 | |
| parent | 822c8ee143a6f8ca78219daa71fe4ad0a9223e21 (diff) | |
| download | pytorch-3749d65a7ef9a327a165556ce2723e43da17c374.tar.gz pytorch-3749d65a7ef9a327a165556ce2723e43da17c374.tar.bz2 pytorch-3749d65a7ef9a327a165556ce2723e43da17c374.zip | |
Create Node2Vec ModuleKeeper
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/18504
Reviewed By: sunnieshang
Differential Revision: D14632091
fbshipit-source-id: d4544866552dc6bcbc7515be9e88cb11e7622a44
Diffstat (limited to 'caffe2')
| -rw-r--r-- | caffe2/operators/text_file_reader.cc | 1 | ||||
| -rw-r--r-- | caffe2/python/text_file_reader.py | 2 |
2 files changed, 2 insertions, 1 deletions
diff --git a/caffe2/operators/text_file_reader.cc b/caffe2/operators/text_file_reader.cc index 4afd689fdc..cbe3d81fd6 100644 --- 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 c351bcdb94..52a1b274f0 100644 --- 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. """ |