Spaces:
Running
on
Zero
Running
on
Zero
File size: 859 Bytes
4893ce0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
"""
Main Training Script
Author: Xiaoyang Wu (xiaoyang.wu.cs@gmail.com)
Please cite our work if the code is helpful to you.
"""
from pointcept.engines.defaults import (
default_argument_parser,
default_config_parser,
default_setup,
)
from pointcept.engines.train import TRAINERS
from pointcept.engines.launch import launch
def main_worker(cfg):
cfg = default_setup(cfg)
trainer = TRAINERS.build(dict(type=cfg.train.type, cfg=cfg))
trainer.train()
def main():
args = default_argument_parser().parse_args()
cfg = default_config_parser(args.config_file, args.options)
launch(
main_worker,
num_gpus_per_machine=args.num_gpus,
num_machines=args.num_machines,
machine_rank=args.machine_rank,
dist_url=args.dist_url,
cfg=(cfg,),
)
if __name__ == "__main__":
main()
|