validation_only=False, gpu=False, light_target=False, ): r"""Main logic.""" with initialize_backends( comm_backend=backend, hosts=hosts, rank=rank, logging_level="INFO", logging_file=os.path.join(output_dir, "mlbench.log"), use_cuda=gpu, seed=42, cudnn_deterministic=False, ckpt_run_dir=ckpt_run_dir, delete_existing_ckpts=not validation_only, ): train_loop( run_id, dataset_dir, ckpt_run_dir, output_dir, validation_only, use_cuda=gpu, light_target=light_target, ) if __name__ == "__main__": task_main(main)
gpu=False, light_target=False, ): r"""Main logic.""" with initialize_backends( comm_backend=backend, hosts=hosts, rank=rank, logging_level="INFO", logging_file=os.path.join(output_dir, "mlbench.log"), use_cuda=gpu, seed=42, cudnn_deterministic=False, ckpt_run_dir=ckpt_run_dir, delete_existing_ckpts=not validation_only, ): train_loop( run_id, dataset_dir, ckpt_run_dir, output_dir, validation_only, use_cuda=gpu, light_target=light_target, ) if __name__ == "__main__": task_main(main, uid="ddp")