- - + Minimal example
- - + How do these methods fit into the broader training?
-
-
+
Required Methods
@@ -298,14 +298,14 @@
- - + training_step
-
-
+
train_dataloader
@@ -313,7 +313,7 @@
- - + Return @@ -325,7 +325,7 @@
-
-
+
configure_optimizers
@@ -333,7 +333,7 @@
- - + Return @@ -350,7 +350,7 @@
-
-
+
Optional Methods
@@ -358,35 +358,35 @@
- - + validation_step
- - + validation_end
- - + test_step
- - + test_end
-
-
+
on_save_checkpoint
@@ -394,7 +394,7 @@
- - + Return @@ -406,7 +406,7 @@
-
-
+
on_load_checkpoint
@@ -414,7 +414,7 @@
- - + Return @@ -426,7 +426,7 @@
-
-
+
val_dataloader
@@ -434,7 +434,7 @@
- - + Return @@ -446,7 +446,7 @@
-
-
+
test_dataloader
@@ -454,7 +454,7 @@
- - + Return @@ -466,7 +466,7 @@
-
-
+
add_model_specific_args
@@ -474,7 +474,7 @@
-
-
+
Return
@@ -731,21 +731,21 @@
- - + Minimal example
- - + How do these methods fit into the broader training?
-
-
+
Required Methods
@@ -753,14 +753,14 @@
- - + training_step
-
-
+
train_dataloader
@@ -768,7 +768,7 @@
- - + Return @@ -780,7 +780,7 @@
-
-
+
configure_optimizers
@@ -788,7 +788,7 @@
- - + Return @@ -805,7 +805,7 @@
-
-
+
Optional Methods
@@ -813,35 +813,35 @@
- - + validation_step
- - + validation_end
- - + test_step
- - + test_end
-
-
+
on_save_checkpoint
@@ -849,7 +849,7 @@
- - + Return @@ -861,7 +861,7 @@
-
-
+
on_load_checkpoint
@@ -869,7 +869,7 @@
- - + Return @@ -881,7 +881,7 @@
-
-
+
val_dataloader
@@ -889,7 +889,7 @@
- - + Return @@ -901,7 +901,7 @@
-
-
+
test_dataloader
@@ -909,7 +909,7 @@
- - + Return @@ -921,7 +921,7 @@
-
-
+
add_model_specific_args
@@ -929,7 +929,7 @@
-
-
+
Return
@@ -2047,7 +2047,7 @@ The
+
diff --git a/LightningModule/methods/index.html b/LightningModule/methods/index.html
index a4457f9f..cb5f8c28 100644
--- a/LightningModule/methods/index.html
+++ b/LightningModule/methods/index.html
@@ -34,7 +34,7 @@
-
+
@@ -42,7 +42,7 @@
-
+
@@ -171,7 +171,7 @@
-
+ @@ -284,21 +284,21 @@- + diff --git a/LightningModule/properties/index.html b/LightningModule/properties/index.html index f26543d8..1e5e67c1 100644 --- a/LightningModule/properties/index.html +++ b/LightningModule/properties/index.html @@ -34,7 +34,7 @@ - + @@ -42,7 +42,7 @@ - + @@ -171,7 +171,7 @@ -+ @@ -296,56 +296,56 @@- + diff --git a/Trainer/Checkpointing/index.html b/Trainer/Checkpointing/index.html index 5dfbbb7b..14232591 100644 --- a/Trainer/Checkpointing/index.html +++ b/Trainer/Checkpointing/index.html @@ -34,7 +34,7 @@ - + @@ -42,7 +42,7 @@ - + @@ -171,7 +171,7 @@ -- - + current_epoch
- - + dtype
- - - experiment + + logger
- - + global_step
- - + gradient_clip_val
- - + on_gpu
- - + trainer
-
-
+
Debugging
@@ -353,7 +353,7 @@
-
-
+
example_input_array
@@ -579,56 +579,56 @@
- - + current_epoch
- - + dtype
- - - experiment + + logger
- - + global_step
- - + gradient_clip_val
- - + on_gpu
- - + trainer
-
-
+
Debugging
@@ -636,7 +636,7 @@
-
-
+
example_input_array
@@ -676,13 +676,21 @@
dtype
Current dtype
-experiment
-An instance of test-tube Experiment which you can use to log anything for tensorboard (subclass of PyTorch SummaryWriter).
+logger
+A reference to the logger you passed into trainer.
+
+ +1
+Trainer(logger=your_logger) +
Call it from anywhere in your LightningModule to add metrics, images, etc... whatever your logger supports.
+Here is an example using the Test-tube logger (which is a wrapper on PyTorch SummaryWriter with versioned folder structure).
@@ -788,7 +796,7 @@ to get a print out of sizes coming into and out of every layer.1 2 -3
self.experiment.add_embedding(...) -self.experiment.log({'val_loss': 0.9}) -self.experiment.add_scalars(...) +3 +4
# if logger is a tensorboard logger or test-tube experiment +self.logger.add_embedding(...) +self.logger.log({'val_loss': 0.9}) +self.logger.add_scalars(...)
-
-
+
example_input_array
@@ -676,13 +676,21 @@
-
-
+
example_input_array
@@ -579,56 +579,56 @@
+ @@ -344,14 +344,14 @@- + diff --git a/Trainer/Distributed training/index.html b/Trainer/Distributed training/index.html index 42aca7fe..08d7b851 100644 --- a/Trainer/Distributed training/index.html +++ b/Trainer/Distributed training/index.html @@ -34,7 +34,7 @@ - + @@ -42,7 +42,7 @@ - + @@ -171,7 +171,7 @@ -- - + Model saving
-
-
+
Restoring training session
@@ -524,14 +524,14 @@
- - + Model saving
- - + Restoring training session @@ -725,7 +725,7 @@ the system isn't different. If you add a layer, for instance, it might not work.
+ @@ -356,63 +356,63 @@- - + Choosing a backend
- - + Distributed and 16-bit precision.
- - + CUDA flags
- - + 16-bit mixed precision
- - + Single-gpu
- - + multi-gpu
- - + Multi-node
- - + Auto-slurm-job-submission
-
-
+
Self-balancing architecture
@@ -573,63 +573,63 @@
- - + Choosing a backend
- - + Distributed and 16-bit precision.
- - + CUDA flags
- - + 16-bit mixed precision
- - + Single-gpu
- - + multi-gpu
- - + Multi-node
- - + Auto-slurm-job-submission
-
-
+
Self-balancing architecture
@@ -1119,7 +1119,7 @@ do this for you. The SlurmCluster can also run a grid search if you pass in a
-
+
diff --git a/Trainer/Logging/index.html b/Trainer/Logging/index.html
index cdb88187..7e3863c5 100644
--- a/Trainer/Logging/index.html
+++ b/Trainer/Logging/index.html
@@ -34,7 +34,7 @@
-
+
@@ -42,7 +42,7 @@
-
+
@@ -79,7 +79,7 @@
-
+
Skip to content
@@ -171,7 +171,7 @@
-
+ @@ -368,57 +368,84 @@- - - Display metrics in progress bar - - - - -
- - - Log metric row every k batches - - - - -
- - - Log metric row every k batches - - - - -
- - - Process position - - - - -
- - - Save a snapshot of all hyperparameters - - - - -
- - - Snapshot code for a training run - - - - -
- - - Tensorboard support + + Setting up logging + + + +
- + + Using loggers + + +
-
-
+
Return
@@ -2047,7 +2047,7 @@ The
+
diff --git a/LightningModule/methods/index.html b/LightningModule/methods/index.html
index a4457f9f..cb5f8c28 100644
--- a/LightningModule/methods/index.html
+++ b/LightningModule/methods/index.html
@@ -34,7 +34,7 @@
-
+
@@ -42,7 +42,7 @@
-
+
@@ -171,7 +171,7 @@
-
-
-
+
Return
@@ -731,21 +731,21 @@