提交 9a243d7a 编写于 作者: Z Zhizhong Su

fix a few bugs related to frame size.

上级 be734a60
...@@ -54,7 +54,7 @@ void DataLayer::copyDataToOutput(Argument& output) { ...@@ -54,7 +54,7 @@ void DataLayer::copyDataToOutput(Argument& output) {
output.setFrameWidth(config_.width()); output.setFrameWidth(config_.width());
} else { } else {
output.setFrameHeight(data_.getFrameHeight()); output.setFrameHeight(data_.getFrameHeight());
output.setFrameHeight(data_.getFrameHeight()); output.setFrameWidth(data_.getFrameWidth());
} }
output.cpuSequenceDims = data_.cpuSequenceDims; output.cpuSequenceDims = data_.cpuSequenceDims;
output.sequenceStartPositions = data_.sequenceStartPositions; output.sequenceStartPositions = data_.sequenceStartPositions;
......
...@@ -245,6 +245,8 @@ int32_t Argument::resizeAndCopyFrom(const Argument& src, ...@@ -245,6 +245,8 @@ int32_t Argument::resizeAndCopyFrom(const Argument& src,
bool useGpu, bool useGpu,
hl_stream_t stream) { hl_stream_t stream) {
dataId = src.dataId; dataId = src.dataId;
frameWidth = src.frameWidth;
frameHeight = src.frameHeight;
if (!src.sequenceStartPositions) { if (!src.sequenceStartPositions) {
// non-sequence input, copy samples directly // non-sequence input, copy samples directly
......
...@@ -245,7 +245,7 @@ message ImageConfig { ...@@ -245,7 +245,7 @@ message ImageConfig {
// The size of input feature map. // The size of input feature map.
required uint32 img_size = 8; required uint32 img_size = 8;
required uint32 img_size_y = 9; optional uint32 img_size_y = 9;
} }
message LayerInputConfig { message LayerInputConfig {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册