Skip to content

Commit

Permalink
add maskrcnn inference example (intel-analytics#2944)
Browse files Browse the repository at this point in the history
* add maskrcnn inference example

* meet pr comments

* add model download url
  • Loading branch information
zhangxiaoli73 committed Oct 28, 2019
1 parent c7f44a2 commit c5c2582
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -88,14 +88,14 @@ class Evaluator[T: ClassTag] private[optim](model: Module[T])(implicit ev: Tenso
vMethods: Array[ValidationMethod[T]]
): Array[(ValidationResult, ValidationMethod[T])] = {

val dummyInput = dataset.takeSample(withReplacement = false, num = 1).head.getInput()
val rdd = ConversionUtils.coalesce(dataset)
val modelBroad = ModelBroadcast[T]().broadcast(rdd.sparkContext,
ConversionUtils.convert(model.evaluate()), dummyInput)
ConversionUtils.convert(model.evaluate()))
val otherBroad = rdd.sparkContext.broadcast(vMethods)


rdd.mapPartitions(miniBatch => {
val localModel = modelBroad.value(false, true, dummyInput)
val localModel = modelBroad.value()
val localMethod = otherBroad.value
miniBatch.map(batch => {
val output = localModel.forward(batch.getInput())
Expand Down

0 comments on commit c5c2582

Please sign in to comment.