Call System.gc for each iteration

ThoughtWorksInc · Jun 1, 2018 · 10b34e4 · 10b34e4
1 parent 2a5efc2
commit 10b34e4
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/benchmark/src/jmh/scala/com/thoughtworks/deeplearning/benchmark/benchmark.scala b/benchmark/src/jmh/scala/com/thoughtworks/deeplearning/benchmark/benchmark.scala
@@ -42,7 +42,7 @@ object benchmark {
   @State(Scope.Benchmark)
   class FourLayer {
 
-    @Param(Array("4"))
+    @Param(Array("8"))
     protected var batchSize: Int = _
 
     @Param(Array("1", "2", "4"))
@@ -170,10 +170,15 @@ object benchmark {
 
     @Benchmark
     final def deepLearningDotScala(): Double = {
-      val (coarseClass, batch) = batches.synchronized {
-        batches.next()
+      try {
+        val (coarseClass, batch) = batches.synchronized {
+          batches.next()
+        }
+        model.train(coarseClass, batch).blockingAwait
+      } finally {
+        System.gc()
+        System.gc()
       }
-      model.train(coarseClass, batch).blockingAwait
     }
 
   }