stmasson commited on
Commit
348bed7
·
verified ·
1 Parent(s): 334164b

Upload train_alizee_coder.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. train_alizee_coder.py +1 -1
train_alizee_coder.py CHANGED
@@ -26,7 +26,7 @@ print("HF_TOKEN verified")
26
 
27
  print(f"Loading dataset nvidia/OpenCodeReasoning...")
28
  try:
29
- dataset = load_dataset("nvidia/OpenCodeReasoning", "split_0", split="train")
30
  dataset = dataset.shuffle(seed=42).select(range(min(DATASET_SIZE, len(dataset))))
31
  print(f"Dataset loaded: {len(dataset)} examples")
32
  except Exception as e:
 
26
 
27
  print(f"Loading dataset nvidia/OpenCodeReasoning...")
28
  try:
29
+ dataset = load_dataset("nvidia/OpenCodeReasoning", "split_0", split="split_0")
30
  dataset = dataset.shuffle(seed=42).select(range(min(DATASET_SIZE, len(dataset))))
31
  print(f"Dataset loaded: {len(dataset)} examples")
32
  except Exception as e: