We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ff3e2dc commit 9f1eadeCopy full SHA for 9f1eade
docs/source/extensions/plugins.rst
@@ -81,7 +81,9 @@ can then be passed into the Trainer directly or via a (custom) accelerator:
81
82
# fully custom accelerator and plugins
83
accelerator = MyAccelerator()
84
- trainer = Trainer(accelerator=accelerator)
+ precision_plugin = MyPrecisionPlugin()
85
+ training_type_plugin = CustomDDPPlugin(accelerator=accelerator, precision_plugin=precision_plugin)
86
+ trainer = Trainer(strategy=training_type_plugin)
87
88
89
The full list of built-in plugins is listed below.
0 commit comments