Skip to content

Commit 5bfa28f

Browse files
authored
Merge pull request #897 from zzsu/bug-fix
fix a few bugs related to frame size.
2 parents 5ac16e5 + 9a243d7 commit 5bfa28f

File tree

3 files changed

+4
-2
lines changed

3 files changed

+4
-2
lines changed

paddle/gserver/layers/DataLayer.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ void DataLayer::copyDataToOutput(Argument& output) {
5454
output.setFrameWidth(config_.width());
5555
} else {
5656
output.setFrameHeight(data_.getFrameHeight());
57-
output.setFrameHeight(data_.getFrameHeight());
57+
output.setFrameWidth(data_.getFrameWidth());
5858
}
5959
output.cpuSequenceDims = data_.cpuSequenceDims;
6060
output.sequenceStartPositions = data_.sequenceStartPositions;

paddle/parameter/Argument.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -245,6 +245,8 @@ int32_t Argument::resizeAndCopyFrom(const Argument& src,
245245
bool useGpu,
246246
hl_stream_t stream) {
247247
dataId = src.dataId;
248+
frameWidth = src.frameWidth;
249+
frameHeight = src.frameHeight;
248250

249251
if (!src.sequenceStartPositions) {
250252
// non-sequence input, copy samples directly

proto/ModelConfig.proto

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -245,7 +245,7 @@ message ImageConfig {
245245

246246
// The size of input feature map.
247247
required uint32 img_size = 8;
248-
required uint32 img_size_y = 9;
248+
optional uint32 img_size_y = 9;
249249
}
250250

251251
message LayerInputConfig {

0 commit comments

Comments
 (0)