Learn practical skills, build real-world projects, and advance your career
Updated 4 years ago
import wandb
from wandb.tensorflow import WandbHook
wandb.init(project="test-wandb", sync_tensorboard=True)!pip install jovian --upgrade --quiet
import jovian
project_name = "wandb-test"[jovian] Attempting to save notebook..
[jovian] Detected Kaggle notebook...
[jovian] Please enter your API key ( from https://jovian.ml/ ):
API KEY: ········
[jovian] Uploading notebook to https://jovian.ml/8dave8/wandb-test
pip install pytorch-lightning
import pytorch_lightning as pl
from pytorch_lightning.loggers import WandbLoggerjovian.commit(project=project_name, environment=None)[jovian] Attempting to save notebook..
[jovian] Detected Kaggle notebook...
[jovian] Uploading notebook to https://jovian.ml/8dave8/wandb-test
Examples
classifier.train(input_fn=train_input_fn, steps=100000, hooks=[WandbHook()])
Save model to wandb
saver = tf.train.Saver()
saver.save(sess, os.path.join(wandb.run.dir, "model.ckpt"))