From 0b0f2c01e1b623abf811465d265efa221d785710 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 16 Jan 2020 10:16:17 -0500 Subject: [PATCH] working on trainer docs --- pytorch_lightning/trainer/trainer.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index b9c0f55b..add1c849 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -360,7 +360,11 @@ class Trainer(TrainerIOMixin, weights_save_path='my/path' ) - # :param str amp_level: Check nvidia docs for level + amp_level (str): The optimization level to use (O1, O2, etc...). + Check nvidia docs for level (https://nvidia.github.io/apex/amp.html#opt-levels) + Example:: + # default used by the Trainer + trainer = Trainer(amp_level='O1') """ # :param int num_sanity_val_steps: How many val steps before a full train loop. # :param int truncated_bptt_steps: Enables multiple backward passes for each batch.