Skip to content

Instantly share code, notes, and snippets.

@aribornstein

aribornstein/lightning_init.py

Created February 14, 2021 15:11
Show Gist options
  • Save aribornstein/8f778e5f23a7011b99575c22ed065424 to your computer and use it in GitHub Desktop.
Save aribornstein/8f778e5f23a7011b99575c22ed065424 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
lightning_init.py
class LitModel(pl.LightningModule):
def __init__(self, encoder: nn.Module, coeff_x: float = 0.2, lr: float = 1e-3)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment