added dataloader, criterion, model definition and began defining training

2024-03-11 20:33:49 +01:00
parent 94ad3d28f1
commit 9fa3574a7d
4 changed files with 201 additions and 0 deletions
@@ -0,0 +1,6 @@
+from __future__ import annotations
+
+from dataloader import VCDADataset  # noqa: F401
+from loss_fn import setup_criterion  # noqa: F401
+
+from model import ResNet18Head  # noqa: F401