marian-nmt/marian
1
1
Fork 0
mirror of https://github.com/marian-nmt/marian.git synced 2024-11-04 14:04:24 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add initial version of validation

Browse Source
This commit is contained in:
Tomasz Dwojak 2016-09-14 15:47:05 +01:00
parent 67e717d366
commit 269936a4a8
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/npz_converter.h
View File

@ -53,13 +53,18 @@ class NpzConverter {
destructed_ = true;
}
void Load(const std::string& key, marian::Tensor& tensor) const {
void Load(const std::string& key, std::vector<float>& data, marian::Shape& shape) const {
auto it = model_.find(key);
if(it != model_.end()) {
NpyMatrixWrapper np(it->second);
tensor.allocate({(int)np.size1(), (int)np.size2()});
std::vector<float> data(np.data(), np.data() + np.size());
tensor.Load(data);
data.clear();
data.resize(np.size());
std::copy(np.data(), np.data() + np.size(), data.begin());
shape.clear();
shape.push_back(np.size1());
shape.push_back(np.size2());
}
else {
std::cerr << "Missing " << key << std::endl;