r'''Wraps ConfigProto for distributed training.
(session_config, *args, **kwargs)
| 441 | |
| 442 | |
| 443 | def wraps_session_config(session_config, *args, **kwargs): |
| 444 | r'''Wraps ConfigProto for distributed training. |
| 445 | ''' |
| 446 | if not session_config: |
| 447 | kwargs.setdefault('allow_soft_placement', True) |
| 448 | session_config = config_pb2.ConfigProto(*args, **kwargs) |
| 449 | session_config.gpu_options.allow_growth = True |
| 450 | session_config.gpu_options.force_gpu_compatible = True |
| 451 | # Horovod: pin GPU to be used to process local rank (one GPU per process) |
| 452 | session_config.gpu_options.visible_device_list = str(HvdContext.get().rank) |
| 453 | return session_config |
| 454 | |
| 455 | def wraps_monitored_training_session(fn): |
| 456 | r'''Decorator to create wrapped monitored training session. |
no test coverage detected