Single GPU process
(local_rank, main_port, cfg, world_size)
| 26 | |
| 27 | |
| 28 | def single_proc_run(local_rank, main_port, cfg, world_size): |
| 29 | """Single GPU process""" |
| 30 | os.environ["MASTER_ADDR"] = "localhost" |
| 31 | os.environ["MASTER_PORT"] = str(main_port) |
| 32 | os.environ["RANK"] = str(local_rank) |
| 33 | os.environ["LOCAL_RANK"] = str(local_rank) |
| 34 | os.environ["WORLD_SIZE"] = str(world_size) |
| 35 | try: |
| 36 | register_omegaconf_resolvers() |
| 37 | except Exception as e: |
| 38 | logging.info(e) |
| 39 | |
| 40 | trainer = instantiate(cfg.trainer, _recursive_=False) |
| 41 | trainer.run() |
| 42 | |
| 43 | |
| 44 | def single_node_runner(cfg, main_port: int): |
no test coverage detected