averaging models scripts

scripts/average.py

@@ -0,0 +1,20 @@
+#!/usr/bin/env python
+
+import sys
+import numpy as np;
+
+average = dict()
+
+for filename in sys.argv[1:-1]:
+    with open(filename, "rb") as mfile:
+        m = np.load(mfile)
+        for k in m:
+            if k not in average:
+                average[k] = m[k]
+            elif average[k].shape == m[k].shape:
+                average[k] += m[k]
+
+for k in average:
+    average[k] /= len(sys.argv[1:-1])
+
+np.savez(sys.argv[-1], **average)

src/decoder/decoder_main.cu

@@ -24,8 +24,6 @@ int main(int argc, char* argv[]) {
   God::Init(argc, argv);
   boost::timer::cpu_timer timer;
   
-  LOG(info) << "Reading input";
-  
   std::string in;
   std::size_t taskCounter = 0;
   
@@ -34,6 +32,8 @@ int main(int argc, char* argv[]) {
   LOG(info) << "Setting number of threads to " << threadCount;
   ThreadPool pool(threadCount);
   std::vector<std::future<History>> results;
+  
+  LOG(info) << "Reading input";
   while(std::getline(std::cin, in)) {
     
     results.emplace_back(