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

protected legacy_pad_, replace DeleteDropout with is_test=True

Browse Source
This commit is contained in:
Song Han 2016-07-29 11:44:55 -07:00
parent 3c989347d8
commit cc46464cf6
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
caffe2/operators/conv_pool_op_base.h
View File

@ -163,7 +163,6 @@ class ConvPoolOpBase : public Operator<Context> {
// exact padding values. This isolates out the padding scheme that are growing // exact padding values. This isolates out the padding scheme that are growing
// unfortunately complex due to implementational differences from different // unfortunately complex due to implementational differences from different
// frameworks. // frameworks.
LegacyPadding legacy_pad_;
int pad_; int pad_;
protected: protected:
@ -176,6 +175,7 @@ class ConvPoolOpBase : public Operator<Context> {
int stride_h_; int stride_h_;
int stride_w_; int stride_w_;
StorageOrder order_; StorageOrder order_;
LegacyPadding legacy_pad_;
inline void ComputeSizeAndPad( inline void ComputeSizeAndPad(
const int in_size, const int stride, const int kernel, const int in_size, const int stride, const int kernel,

3
caffe2/python/caffe_translator_test.py
View File

@ -56,9 +56,8 @@ if __name__ == '__main__':
.read() .read()
) )
net, pretrained_params = caffe_translator.TranslateModel( net, pretrained_params = caffe_translator.TranslateModel(
caffenet, caffenet_pretrained caffenet, caffenet_pretrained, is_test=True
) )
caffe_translator.DeleteDropout(net)
with open('data/testdata/caffe_translator/' with open('data/testdata/caffe_translator/'
'bvlc_reference_caffenet.translatedmodel', 'bvlc_reference_caffenet.translatedmodel',
'w') as fid: 'w') as fid: