Add the size thing

c-salomonsen · c-salomonsen · commit 20faa24e148d · 2025-02-08T20:18:54.000+01:00
diff --git a/main.py b/main.py
@@ -46,6 +46,7 @@ def main():
         data_path=args.datafolder,
         transform=transform,
         download=args.download_data,
+        val_size=args.val_size,
     )
 
     metrics = MetricWrapper(*args.metric, num_classes=traindata.num_classes)

Original file line number	Diff line number	Diff line change
`@@ -46,6 +46,7 @@ def main():`
`46`	`46`	`data_path=args.datafolder,`
`47`	`47`	`transform=transform,`
`48`	`48`	`download=args.download_data,`
	`49`	`+ val_size=args.val_size,`
`49`	`50`	`)`
`50`	`51`
`51`	`52`	`metrics = MetricWrapper(*args.metric, num_classes=traindata.num_classes)`