Skip to content

Instantly share code, notes, and snippets.

@AbdelmajidB
Created November 25, 2019 16:28
Show Gist options
  • Save AbdelmajidB/6eb3750aba682a3481725a1f7752b455 to your computer and use it in GitHub Desktop.
Save AbdelmajidB/6eb3750aba682a3481725a1f7752b455 to your computer and use it in GitHub Desktop.
while (Iter.hasNext()) {
DataSet next = Iter.next();
INDArray out2d = modelT.outputSingle(next.getFeatures()).permute(0,2,3,1).dup().reshape('c',height*width,1);
INDArray labels2d = next.getLabels().permute(0,2,3,1).dup().reshape('c',height*width,1);
if(k==0) {
e.eval(labels2d, out2d);
log.info(e.stats());
}
k++;
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment