ayulockin

# Imports
import tensorflow as tf
from tensorflow.keras.layers import *
from tensorflow.keras.models import *

import numpy as np

import wandb
from wandb.keras import WandbCallback

ayulockin

import wandb

import tensorflow as tf
from keras_cv import bounding_box

class WandbTablesBuilder:
    """
    Utility class that contains useful methods to create W&B Tables,
    and log it to W&B.
    """

ayulockin

import tensorflow as tf

import wandb
from wandb.keras import WandBMetricsLogger

with wandb.init(project="mnist", job_type="dev-wandb-metrics-logger"):

    fashion_mnist = tf.keras.datasets.fashion_mnist

    (train_images, train_labels), (test_images, test_labels) = fashion_mnist.load_data()

ayulockin

import tensorflow as tf

import wandb
from wandb.keras import (
    # WandBMetricsLogger,
    WandbModelCheckpoint,
    # WandbGradientLogger,
    # ModelLogger,
    # FLOPsLogger,
)

ayulockin

# Demonstrative changes to this file: https://github.com/Lightning-AI/lit-gpt/blob/main/finetune/lora.py

from lightning.pytorch.loggers import WandbLogger


def setup(
    data_dir: Path = Path("data/alpaca"),
    checkpoint_dir: Path = Path("checkpoints/stabilityai/stablelm-base-alpha-3b"),
    out_dir: Path = Path("out/lora/alpaca"),
    precision: Optional[str] = None,