OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Fix TensorProtosDBInput AttributeError (#32274)

Browse Source 
Summary:
https://github.com/pytorch/pytorch/issues/6794
Pull Request resolved: https://github.com/pytorch/pytorch/pull/32274

Differential Revision: D19621889

Pulled By: ezyang

fbshipit-source-id: 1bdd042b6421a2798c7f1e9030dfc6dfc1246989
This commit is contained in:
Deepali Chourasia 2020-01-29 12:00:49 -08:00 committed by Facebook Github Bot
parent 8693164acb
commit e84f9d9d0c
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
caffe2/python/model_helper.py
View File

@ -5,7 +5,8 @@ from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
from caffe2.python import core, scope, workspace, helpers
from caffe2.python import core, scope, workspace
from caffe2.python.helpers.db_input import db_input
from caffe2.python.modeling import parameter_info
from caffe2.python.modeling.parameter_sharing import (
parameter_sharing_context,
@ -414,7 +415,7 @@ class ModelHelper(object):
"""You cannot pass reader to model_helper.TensorProtosDBInput.
Use model.net.TensorProtosDBInput instead to create the op."""
return helpers.db_input.db_input(
return db_input(
self, blob_out, batch_size, db, db_type, **kwargs)
def GetDevices(self):