← Về thư mục
skills/pytorch-lightning/scripts/template_lightning_module.py
- TemplateLightningModule · class · L16-L195 — class TemplateLightningModule(L.LightningModule)
- init · method · L26-L48 — def init( self, learning_rate: float = 0.001, hidden_dim: int = 256, dropout: float = 0.1, )
- forward · method · L50-L60 — def forward(self, x)
- training_step · method · L62-L88 — def training_step(self, batch, batch_idx)
- validation_step · method · L90-L110 — def validation_step(self, batch, batch_idx)
- test_step · method · L112-L132 — def test_step(self, batch, batch_idx)
- predict_step · method · L134-L149 — def predict_step(self, batch, batch_idx, dataloader_idx=0)
- configure_optimizers · method · L151-L183 — def configure_optimizers(self)
- on_train_epoch_end · method · L187-L190 — def on_train_epoch_end(self)
- on_validation_epoch_end · method · L192-L195 — def on_validation_epoch_end(self)