Category "pytorch-lightning"

Access all batch outputs at the end of epoch in callback with pytorch lightning

The documentation for the on_train_epoch_end, https://pytorch-lightning.readthedocs.io/en/stable/extensions/callbacks.html#on-train-epoch-end, states: To acces

Can't set attribute for Pytorch LightningModule

I am trying to import initiate a LightningModule class module, but for some reasons i unable to set hparams. However if i change the name to hparams2 it suddenl

Pytorch lightning resuming from checkpoint with new data

I'm wanting to continue the training process for a model using new data. I understand that you can continue training a Pytorch Lightning model e.g. pl.Trainer(m

How to import TensorMetric in pytorch lighting?

I want to import TensorMetric : from pytorch_lightning.metrics.metric import TensorMetric The program throws an exception: ModuleNotFoundError: No module name

load_from_checkpoint fails after transfer learning a LightningModule

I try to transfer learn a LightningModule. The relevant part of the code is this: class DeepFilteringTransferLearning(pl.LightningModule): def __init__(self

PyTorch Lightning training console output is weird

When training a PyTorch Lightning model in a Jupyter Notebook, the console log output is awkward: Epoch 0: 100%|███████