Skip to content

Lightsaber version 0.3

Latest
Latest
Compare
Choose a tag to compare
@pchakraborty1 pchakraborty1 released this 05 Aug 13:07
· 19 commits to main since this release
lightsaber_v0.3
54bd849
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

Highlights

  • upgraded pytorch_lightning to version 1.6.4
  • updated test script, api largely backported
  • Updated example for LSTM/HistGBT

Changes in API:

  • pl_trainer.run_training_with_mlflow:

    • Call Signature updated

      • dataloaders can be now passed directly here instead of in wrapped model
      • instead of instantiated trainer, arguments supported by pytorch lightning Trainer is passed as Namespace/ArgumentParser

    • Return type updated. validation/test input and outputs (e.g. y_val, y_pred) are now numpy not torch tensors

Assets 3
Loading