jskvrna
/

hoho

Model card Files Files and versions

jskvrna commited on Jun 5, 2025

Commit

3c656e6

·

1 Parent(s): a55a0be

x

Files changed (1) hide show

train.py +1 -1

train.py CHANGED Viewed

@@ -48,7 +48,7 @@ os.makedirs(args.results_dir, exist_ok=True)
 #ds = load_dataset("usm3d/hoho25k", cache_dir="/media/skvrnjan/sd/hoho25k/", trust_remote_code=True)
 ds = load_dataset("usm3d/hoho25k", cache_dir="/mnt/personal/skvrnjan/hoho25k/", trust_remote_code=True)
-#ds = ds.shuffle()
 scores_hss = []
 scores_f1 = []

 #ds = load_dataset("usm3d/hoho25k", cache_dir="/media/skvrnjan/sd/hoho25k/", trust_remote_code=True)
 ds = load_dataset("usm3d/hoho25k", cache_dir="/mnt/personal/skvrnjan/hoho25k/", trust_remote_code=True)
+ds = ds.shuffle()
 scores_hss = []
 scores_f1 = []