Overview
Plugin is an abstraction to inject features into the training components passed by the user. This plugin will perform wrapping/transformation to convert the training components to ones which are accelerated/modified.
Wanna track the development progress? Take a look at
Goal
The Plugin class should be implemented to achieve some transformation for demonstration purpose. For the current stage, we can convert the training components with Gemini, i.e. write a plugin for PyTorch DDP.
Overview
Pluginis an abstraction to inject features into the training components passed by the user. This plugin will perform wrapping/transformation to convert the training components to ones which are accelerated/modified.Wanna track the development progress? Take a look at
Goal
The
Pluginclass should be implemented to achieve some transformation for demonstration purpose. For the current stage, we can convert the training components with Gemini, i.e. write a plugin for PyTorch DDP.