o  iy@s~ddlmZddlmZmZddlmZmZddlm Z ddl m Z m Z GdddZ Gd d d e e ZGd d d e e Zd S)) annotations)MappingSequence) TYPE_CHECKINGAny)optional_import)TensorBoardImageHandlerTensorBoardStatsHandlerc@seZdZdZdddZdS)ClearMLHandleraU Base class for the handlers to log everything to ClearML. For more details of ClearML usage, please refer to: https://clear.ml/docs/latest/docs/references/sdk/task Usage example is available in the tutorial: https://github.com/Project-MONAI/tutorials/blob/master/3d_segmentation/unet_segmentation_3d_ignite.ipynb. project_name str | None task_name output_uri str | booltagsSequence[str] | Nonereuse_last_task_idboolcontinue_last_taskauto_connect_frameworks&bool | Mapping[str, bool | str | list]auto_connect_arg_parserbool | Mapping[str, bool]returnNonec CsVtrddl} ntd\} } | jr| j|_dS| jj||||||||d|_dS) Args: project_name: ClearML project name, default to 'MONAI'. task_name: ClearML task name, default to 'monai_experiment'. output_uri: The default location for output models and other artifacts, default to 'True'. tags: Add a list of tags (str) to the created Task, default to 'None'. reuse_last_task_id: Force a new Task (experiment) with a previously used Task ID, default to 'True'. continue_last_task: Continue the execution of a previously executed Task (experiment), default to 'False'. auto_connect_frameworks: Automatically connect frameworks, default to 'True'. auto_connect_arg_parser: Automatically connect an argparse object to the Task, default to 'True'. rNclearmlr rrrrrrr)rrrTask current_taskZ clearml_taskinit) selfr rrrrrrrr_r$a/home/dell461/cl/sdc2/last_ska_mid/HISourceFinder-master-l/src/monai/handlers/clearml_handlers.py__init__!s    zClearMLHandler.__init__N)r r rr rrrrrrrrrrrrrr__name__ __module__ __qualname____doc__r&r$r$r$r%r s r c@,eZdZdZ        ddddZdS)ClearMLStatsHandlera< Class to write tensorboard stats by inheriting TensorBoardStatsHandler class. Everything from Tensorboard is logged automatically to ClearML. Usage example is available in the tutorial: https://github.com/Project-MONAI/tutorials/blob/master/3d_segmentation/unet_segmentation_3d_ignite.ipynb. MONAImonai_experimentTNFr r rrrrrrrrrrrrargsrkwargsrrc O8tj|||||||||d tj|g| Ri| dSrrN)r r&r r"r rrrrrrrr0r1r$r$r%r&Zs zClearMLStatsHandler.__init__r.r/TNTFTTr r rr rrrrrrrrrrrrr0rr1rrrr'r$r$r$r%r-O r-c@r,)ClearMLImageHandlera8 This class inherits all functionality from TensorBoardImageHandler class. Everything from Tensorboard is logged automatically to ClearML. Usage example is available in the tutorial: https://github.com/Project-MONAI/tutorials/blob/master/3d_segmentation/unet_segmentation_3d_ignite.ipynb. r.r/TNFr r rrrrrrrrrrrrr0rr1rrc Or2r3)r r&r r4r$r$r%r&s zClearMLImageHandler.__init__r5r6r'r$r$r$r%r8r7r8N) __future__rcollections.abcrrtypingrr monai.utilsrtensorboard_handlersr r r r-r8r$r$r$r%s  93