I am having a problem testing the Benchmark program. I have modified the file Benchmark.java to run the process of creating and training a neural network two times. The first time it takes around 960ms (warmup and actual) while the second time it takes around 4.2ms. I also can observe in the system monitor (Oracle Linux) how it uses all the cores in the first running while it only uses one in the second call to the function (4.5ms). I have attached a file where you can see the simple modification to the Benchmark.java file.
Thank you advance