Update train.py

CVRL · Jul 9, 2024 · 2f95ca9 · 2f95ca9
1 parent 82da5ac
commit 2f95ca9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -105,7 +105,7 @@
 ##################################################################
 ############### Dataloader for train and test data ###############
 ##################################################################
-all_classes = ['real','fake']
+all_classes = ['notcl','tcl']
 class_assgn = {c: i for i, c in enumerate(all_classes)}
 
 dataseta = datasetLoader(args.csvPath, args.datasetPath, args.method, train_test='train', c2i=class_assgn)