Got resnet working - doing some more investigation to find bugs.

src/kbmod_ml/data_sets/kbmod_stamps.py

@@ -28,7 +28,7 @@ def __init__(self, config, split: str):
 
         cols = []
 
-        for c in ["mean", "median"]:
+        for c in ["mean"]:
             cols.append(coadd_type_to_column[c])
 
         self.active_columns = np.array(cols)

src/kbmod_ml/default_config.toml

@@ -1,9 +1,9 @@
 [data_set]
 name = "kbmod_ml.data_sets.kbmod_stamps.KbmodStamps"
 
-train_split = 0.8
+train_size = 0.8
 
-validation_split = 0.2
+validate_size = 0.2
 
 [kbmod_ml]
 # The file name of the true positive samples

train_model.ipynb

@@ -100,9 +100,21 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "from fibad import Fibad\n",
+    "\n",
+    "fibad_instance = Fibad(config_file=\"./user_config.toml\")\n",
+    "\n",
     "# Change the model to the resnet50 model and attempt to train\n",
     "fibad_instance.config[\"model\"][\"name\"] = \"kbmod_ml.models.resnet50.RESNET50\"\n",
-    "fibad_instance.config[\"data_loader\"][\"batch_size\"] = 1\n",
+    "fibad_instance.config[\"train\"][\"epochs\"] = 2"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
     "fibad_instance.train()"
    ]
   },