frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

codemode change missing [from D13586737]

Browse Source 
Summary: as title

Reviewed By: jerryzh168

Differential Revision: D15327669

fbshipit-source-id: e262dacb097e91475b1925ec40b375ec6722ad5a
This commit is contained in:
Du Tran
2019-05-13 20:38:08 -07:00
committed by Facebook Github Bot
parent e8fb5f35f0
commit 8e9692df27
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
caffe2/video/video_input_op.h
View File

@ -484,8 +484,10 @@ VideoInputOp<Context>::VideoInputOp(
label_shape[1] = num_of_class_;
ReinitializeTensor(&prefetched_label_, label_shape, at::dtype<int>().device(CPU));
} else {
prefetched_label_.Resize(
vector<int64_t>(1, batch_size_ * clip_per_video_ * multi_crop_count_));
ReinitializeTensor(
&prefetched_label_,
vector<int64_t>(1, batch_size_ * clip_per_video_ * multi_crop_count_),
at::dtype<int>().device(CPU));
}
ReinitializeTensor(&prefetched_video_id_, vector<int64_t>(1, batch_size_ * clip_per_video_ * multi_crop_count_), at::dtype<int>().device(CPU));