(args)
| 85 | |
| 86 | |
| 87 | def benchmark_train(args): |
| 88 | cfg = setup(args) |
| 89 | model = build_model(cfg) |
| 90 | logger.info("Model:\n{}".format(model)) |
| 91 | if comm.get_world_size() > 1: |
| 92 | model = DistributedDataParallel( |
| 93 | <<<<<<< HEAD |
| 94 | model, device_ids=[comm.get_local_rank()], broadcast_buffers=False, |
| 95 | find_unused_parameters=True |
| 96 | ======= |
| 97 | model, device_ids=[comm.get_local_rank()], broadcast_buffers=False, find_unused_parameters=True |
| 98 | >>>>>>> 7b936afd5b423c3188687d8b529a984bed528a87 |
nothing calls this directly
no test coverage detected