Train on hardhat dataset

hardhat-train.py

from rikai.pytorch.data import Dataset
  
def train(
  name: str,
  uri: Dataset,
  model_type: str,
  epochs: int = 150,
  batch_size: int = 4,
  num_workers: int = 4,
  lr: float = 0.02,
  weight_decay: float = 1e-4,
):
  # Training loop

train(
  "hardhat",
  data=Dataset(
    "datasets/hardhat",
    columns=["image", "annotations"],
    filters=("split = 'train'"),
  )
  model_type="ssd",
)