managers

Class

BaseManager

class gedml.launcher.managers.base_manager.BaseManager(trainer, tester, recorder, objects_dict, device=None, schedulers=None, gradclipper=None, samplers=None, collatefns=None, is_resume=False, is_distributed=False, device_wrapper_type='DP', dist_port=23456, world_size=None, phase='train', primary_metric=['test', 'recall_at_1'], to_device_list=['models', 'collectors'], to_wrap_list=['models'], patience=10)[source]

Bases: object

Manager all modules and computation devices. Support three kinds of computation:

  1. DataParallel (single machine)

  2. DistributedDataParallel (single machine)

  3. DistributedDataParallel (multi machines)