()
| 242 | |
| 243 | |
| 244 | def main(): |
| 245 | for world_size in range(2, args.max_world_size): |
| 246 | delays = [] |
| 247 | for batch in [True, False]: |
| 248 | tik = time.time() |
| 249 | mp.spawn( |
| 250 | run_worker, |
| 251 | args=(world_size, args.num_episode, batch), |
| 252 | nprocs=world_size, |
| 253 | join=True |
| 254 | ) |
| 255 | tok = time.time() |
| 256 | delays.append(tok - tik) |
| 257 | |
| 258 | print(f"{world_size}, {delays[0]}, {delays[1]}") |
| 259 | |
| 260 | |
| 261 | if __name__ == '__main__': |