              section=section,
              title="mask_type",
              datatype=str,
              default="extended",
              choices=PluginLoader.get_available_extractors("mask", add_none=True),
              group="mask",
              gui_radio=True,
              info="The mask to be used for training. If you have selected 'Learn Mask' or "
                   "'Penalized Mask Loss' you must select a value other than 'none'. The required "
                   "mask should have been selected as part of the Extract process. If it does not "
 
</s> remove             choices=["face", "legacy"],
 </s> add             choices=["face", "head", "legacy"], </s> remove #: lib/cli/args.py:890
 </s> add #: lib/cli/args.py:902 </s> remove             choices=PluginLoader.get_available_extractors("mask", add_none=True) + ["predicted"],
 </s> add             choices=PluginLoader.get_available_extractors("mask",
                                                          add_none=True,
                                                          extend_plugin=True) + ["predicted"], </s> remove #: lib/cli/args.py:792
 </s> add #: lib/cli/args.py:804
            choices=PluginLoader.get_available_extractors("mask",
                                                          add_none=True, extend_plugin=True),
              section=section,
              title="mask_type",
              datatype=str,
              default="extended",
            choices=PluginLoader.get_available_extractors("mask",
                                                          add_none=True, extend_plugin=True),
              group="mask",
              gui_radio=True,
              info="The mask to be used for training. If you have selected 'Learn Mask' or "
                   "'Penalized Mask Loss' you must select a value other than 'none'. The required "
                   "mask should have been selected as part of the Extract process. If it does not "
keep keep keep keep replace keep keep keep keep keep
 Masks updates
  - Add head centering support to training
  - Update helptext/tooltips 
plugins/train/_config.py