ssl_tools.models.nets.wisenet

Classes

Module Contents

class ssl_tools.models.nets.wisenet.WiseNet(learning_rate=0.0001)

Bases: pytorch_lightning.LightningModule

Parameters:: learning_rate (float)

_common_step(batch, batch_idx)

configure_optimizers()

forward(x)

predict_step(batch, batch_idx)

test_step(batch, batch_idx)

training_step(batch, batch_idx)

validation_step(batch, batch_idx)