bump version 🚀

gradsflow · Jan 14, 2022 · 2ed6d3b · 2ed6d3b
1 parent dd81035
commit 2ed6d3b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/examples/src/models/hello_world.py b/examples/src/models/hello_world.py
@@ -18,7 +18,13 @@
 from torchvision import transforms as T
 
 from gradsflow import AutoDataset, Model
-from gradsflow.callbacks import CSVLogger, EmissionTrackerCallback, ModelCheckpoint
+from gradsflow.callbacks import (
+    CometCallback,
+    CSVLogger,
+    EmissionTrackerCallback,
+    ModelCheckpoint,
+    WandbCallback,
+)
 from gradsflow.data.common import random_split_dataset
 
 # Replace dataloaders with your custom dataset and you are all set to train your model
@@ -40,6 +46,7 @@
     ModelCheckpoint(),
     EmissionTrackerCallback(),
     # CometCallback(offline=True),
+    WandbCallback(),
 ]
 
 if __name__ == "__main__":

diff --git a/gradsflow/__init__.py b/gradsflow/__init__.py
@@ -28,4 +28,4 @@
 from gradsflow.tuner.automodel import AutoModelV2
 from gradsflow.tuner.tuner import Tuner
 
-__version__ = "0.0.8.dev1"
+__version__ = "0.0.8"