relax regexp for finding weights by name to account for name_scope

transcranial · Dec 29, 2017 · 5bf621d · 5bf621d
1 parent b4ca58c
commit 5bf621d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/Model.js b/src/Model.js
@@ -390,7 +390,7 @@ export default class Model {
     if (weightNames && weightNames.length) {
       const weights = weightNames.map(weightName => {
         const weightDef = _.find(this.modelWeights, w => {
-          const weightRE = new RegExp(`^${weightName}`)
+          const weightRE = new RegExp(`^.*${weightName}`)
           return weightRE.test(w.weightName)
         })