Model Training

In [ ]:

import os
import sys
import warnings
from pathlib import Path

warnings.filterwarnings("ignore")

os.chdir("../../")
# sys.path.append()

In [ ]:

from gradsflow.data.image import image_dataset_from_directory
from gradsflow.core.data import AutoDataset
from gradsflow import Model
import timm

import torch
import torchvision

In [ ]:

data_dir = "/Users/aniket/personal/gradsflow/gradsflow/data"  # replace with your filepath

In [ ]:

train_data = image_dataset_from_directory(
    f"{data_dir}/hymenoptera_data/train/",
    image_size=(96, 96),
    num_workers=None,
    transform=True,
    batch_size=4,
    shuffle=True,
)

val_data = image_dataset_from_directory(
    f"{data_dir}/hymenoptera_data/val/",
    image_size=(96, 96),
    num_workers=None,
    transform=True,
)

train_dataset = train_data["ds"]
train_dl = train_data["dl"]
val_dl = val_data["dl"]
num_classes = len(train_dataset.classes)
autodataset = AutoDataset(train_dl, num_classes=num_classes)

In [ ]:

cnn = timm.create_model("ssl_resnet18", pretrained=True, num_classes=2)
model = Model(cnn, "adam")

In [ ]:

model.fit(
    autodataset,
    epochs=10,
    steps_per_epoch=2,
)

Learning...  ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━    100%  0:00:00 ┏━━━━━━━┳━━━━━━━━━━━━┓
                                                                       ┃ epoch ┃ train/loss ┃
                                                                       ┡━━━━━━━╇━━━━━━━━━━━━┩
                                                                       │ 9     │  0.956     │
                                                                       └───────┴────────────┘

Finished Training

Out[ ]:

Tracker(max_epochs=10, epoch=9, steps_per_epoch=2, train=TrackingValues(loss=0.9556465220246099, steps=3, step_loss=None), val=TrackingValues(loss=None, steps=None, step_loss=None))

In [ ]:

model.fit(
    autodataset,
    epochs=12,
    steps_per_epoch=2,
)

Learning...  ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╺━━━━━━━━━     75%  -:--:-- ┏━━━━━━━┳━━━━━━━━━━━━┓
                                                                       ┃ epoch ┃ train/loss ┃
                                                                       ┡━━━━━━━╇━━━━━━━━━━━━┩
                                                                       │ 9     │  0.375     │
                                                                       └───────┴────────────┘
Training...  ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━      0%  -:--:-- ┏━━━━━━━┳━━━━━━━━━━━━┓
                                                                       ┃ epoch ┃ train/loss ┃
                                                                       ┡━━━━━━━╇━━━━━━━━━━━━┩
                                                                       │ 9     │  0.375     │
                                                                       └───────┴────────────┘

---

Last update: September 26, 2021