Merge pull request #119 from stanford-oval/wip/training-progress

training: fix computing progress for contextual

lib/training/decanlp.js

@@ -68,7 +68,11 @@ module.exports = class DecaNLPTrainingJob extends BaseTrainingJob {
 
         this.metrics = {};
         await execCommand(this, 'decanlp', args, (line) => {
-            const match = /train_[a-z]+:([0-9]+)\/[0-9]+:val_[a-z]+:(.*)$/.exec(line);
+            // the line we are looking for has the form:
+            // ...:train_contextual_almond:70000/100000:val_deca:...
+            // or
+            // ...:train_almond:70000/100000:val_deca:...
+            const match = /train_[a-z_]+:([0-9]+)\/[0-9]+:val_[a-z_]+:(.*)$/.exec(line);
             if (match === null)
                 return;
             this.progress = parseFloat(match[1])/this._config.train_iterations;