Train avg_cost: | Pass | V2(GPU) | Fluid(GPU) | |--------------- | --------------- | ------------------- | |0 | 4.226966 | 4.16706371 | |1 | 3.779494 | 3.80107117 | |2 | 3.467989 | 3.51753068 | |3 | 3.296482 | 3.30209255 | |4 | 3.226758 | 3.19562411 | Test avg_cost: | Pass | V2(GPU) | Fluid(GPU) | |--------------- | --------------- | ------------------- | |0 | 3.830757 | 3.843338 | |1 | 3.587178 | 3.665707 | |2 | 3.371065 | 3.390050 | |3 | 3.323481 | 3.319620 | |3 | 3.299066 | 3.293895 |