From 1b12cb83dcce889c01454e60fdfec35a28a3eaaf Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 16 Jan 2020 10:14:29 -0500 Subject: [PATCH] working on trainer docs --- pytorch_lightning/trainer/trainer.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index 74faa1bc..b9c0f55b 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -355,7 +355,10 @@ class Trainer(TrainerIOMixin, # if checkpoint callback used, then overrides the weights path # **NOTE: this saves weights to some/path NOT my/path checkpoint_callback = ModelCheckpoint(filepath='some/path') - trainer = Trainer(checkpoint_callback=checkpoint_callback, weights_save_path='my/path') + trainer = Trainer( + checkpoint_callback=checkpoint_callback, + weights_save_path='my/path' + ) # :param str amp_level: Check nvidia docs for level """