mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
27c7158166
Summary: There is a module called `2to3` which you can target for future specifically to remove these, the directory of `caffe2` has the most redundant imports: ```2to3 -f future -w caffe2``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/45033 Reviewed By: seemethere Differential Revision: D23808648 Pulled By: bugra fbshipit-source-id: 38971900f0fe43ab44a9168e57f2307580d36a38
18 lines
381 B
Python
18 lines
381 B
Python
## @package db_input
|
|
# Module caffe2.python.helpers.db_input
|
|
|
|
|
|
|
|
|
|
|
|
def db_input(model, blobs_out, batch_size, db, db_type):
|
|
dbreader_name = "dbreader_" + db
|
|
dbreader = model.param_init_net.CreateDB(
|
|
[],
|
|
dbreader_name,
|
|
db=db,
|
|
db_type=db_type,
|
|
)
|
|
return model.net.TensorProtosDBInput(
|
|
dbreader, blobs_out, batch_size=batch_size)
|