Slide 67
Slide 67 text
template void FMAudio8Layer::DataLayerSetUp(
const vector*>& bottom,const vector*>& top){
CHECK_GT(batch_size_*size_,0)<<"oops";
labels_.resize(batch_size_,0);
auto &gen=softfm::training_data_generator::get();
data_[pos_]=std::move(gen.get_pixels());
vector label_shape(1,batch_size_);
top[0]->Reshape(batch_size_,channels_,height_,width_);
top[1]->Reshape(label_shape);
added_data_.Reshape(batch_size_,channels_,height_,width_);
added_label_.Reshape(label_shape);
added_data_.cpu_data();
added_label_.cpu_data();
}
template void FMAudio8Layer::Forward_cpu(
const vector*>& bottom,const vector*>& top){
top[0]->Reshape(batch_size_,channels_,height_,width_);
top[1]->Reshape(batch_size_,1,1,1);
top[0]->set_cpu_data(data_[pos_].data());
top[1]->set_cpu_data(labels_.data());
pos_=(pos_+1)%n_;
auto &gen=softfm::training_data_generator::get();
data_[pos_]=std::move(gen.get_pixels());
}
ੜ͞ΕͨFMԻݯͷεϖΫτϧը૾Λर͏ೖྗ