Fix so nucleus training doesn’t ignore --subset

samples/nucleus/nucleus.py

@@ -252,16 +252,16 @@ class NucleusDataset(utils.Dataset):
 #  Training
 ############################################################
 
-def train(model):
+def train(model, dataset_dir, subset):
     """Train the model."""
     # Training dataset.
     dataset_train = NucleusDataset()
-    dataset_train.load_nucleus(args.dataset, "train")
+    dataset_train.load_nucleus(dataset_dir, subset)
     dataset_train.prepare()
 
     # Validation dataset
     dataset_val = NucleusDataset()
-    dataset_val.load_nucleus(args.dataset, "val")
+    dataset_val.load_nucleus(dataset_dir, "val")
     dataset_val.prepare()
 
     # Image augmentation
@@ -481,7 +481,7 @@ if __name__ == '__main__':
 
     # Train or evaluate
     if args.command == "train":
-        train(model)
+        train(model, args.dataset, args.subset)
     elif args.command == "detect":
         detect(model, args.dataset, args.subset)
     else: