Shortcuts

IPUAccelerator

class pytorch_lightning.accelerators.IPUAccelerator(precision_plugin, training_type_plugin)[source]

Bases: pytorch_lightning.accelerators.accelerator.Accelerator

Accelerator for IPUs.

Parameters
  • precision_plugin (Optional[PrecisionPlugin]) –

    the plugin to handle precision-specific parts

    Deprecated since version The: precision_plugin parameter has been deprecated and will be removed soon. Pass the precision plugin as a parameter to the TrainingTypePlugin instead.

  • training_type_plugin (TrainingTypePlugin) – the plugin to handle different training routines

static auto_device_count()[source]

Get the devices when set to auto.

Return type

int

get_device_stats(device)[source]

IPU device stats aren’t supported yet.

Return type

Dict[str, Any]

setup_optimizers(trainer)[source]
Raises

MisconfigurationException – If multiple optimizers are provided.

Return type

None

Read the Docs v: latest
Versions
latest
stable
1.5.3
1.5.2
1.5.1
1.5.0
1.4.9
1.4.8
1.4.7
1.4.6
1.4.5
1.4.4
1.4.3
1.4.2
1.4.1
1.4.0
1.3.8
1.3.7
1.3.6
1.3.5
1.3.4
1.3.3
1.3.2
1.3.1
1.3.0
1.2.10
1.2.8
1.2.7
1.2.6
1.2.5
1.2.4
1.2.3
1.2.2
1.2.1
1.2.0
1.1.8
1.1.7
1.1.6
1.1.5
1.1.4
1.1.3
1.1.2
1.1.1
1.1.0
1.0.8
1.0.7
1.0.6
1.0.5
1.0.4
1.0.3
1.0.2
1.0.1
1.0.0
0.10.0
0.9.0
0.8.5
0.8.4
0.8.3
0.8.2
0.8.1
0.8.0
0.7.6
0.7.5
0.7.4
0.7.3
0.7.2
0.7.1
0.7.0
0.6.0
0.5.3
0.4.9
ipynb-update
docs-search
Downloads
html
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.