Browse Source

Added continue statement to as that is what is in the original python file and got rid off cast error

tags/v0.10
Pranov Duggasani 6 years ago
parent
commit
29434d3d99
2 changed files with 3 additions and 3 deletions
  1. +1
    -1
      src/TensorFlowNET.Core/Framework/meta_graph.py.cs
  2. +2
    -2
      src/TensorFlowNET.Core/Train/Optimizer.cs

+ 1
- 1
src/TensorFlowNET.Core/Framework/meta_graph.py.cs View File

@@ -134,7 +134,7 @@ namespace Tensorflow
}
break;
default:
throw new NotImplementedException("import_scoped_meta_graph_with_return_elements");
continue;
}
}
}


+ 2
- 2
src/TensorFlowNET.Core/Train/Optimizer.cs View File

@@ -199,8 +199,8 @@ namespace Tensorflow

if (!tf.context.executing_eagerly())
{
var train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP) as List<object>;
if (!train_op.Contains(apply_updates))
var train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP) as List<ITensorOrOperation>;
if (train_op != null && train_op.Contains(apply_updates))
train_op.Add(apply_updates);
}



Loading…
Cancel
Save