res.txt

MPI Process starting with total ranks =  2
0 There are  1  Batches
Batch Size 320
1 [0.11807764] -0.24550347778422843 0.15409326553344727
MPI Process starting with total ranks =  3
0 There are  2  Batches
Batch Size 160
Batch Size 160
2 [0.10522217] -0.13355623144788376 0.07605195045471191
1 [0.10238032] -0.12058970591850443 0.07641720771789551
MPI Process starting with total ranks =  4
0 There are  3  Batches
Batch Size 107
Batch Size 107
Batch Size 106
3 [0.10028532] -0.0905654650858735 0.05823993682861328
2 [0.09832112] -0.0779120098992528 0.05859518051147461
1 [0.09022694] -0.08461628977344444 0.05874991416931152
MPI Process starting with total ranks =  5
0 There are  4  Batches
Batch Size 80
Batch Size 80
Batch Size 80
Batch Size 80
3 [0.08820554] -0.060083476521544964 0.03981184959411621
2 [0.09592697] -0.06543847372369281 0.04426407814025879
4 [0.09763171] -0.0674572883799123 0.044415950775146484
1 [0.08873866] -0.05223308043235639 0.04548501968383789