Spaces:

phuong-d-h-nguyen
/

test-public-space

Runtime error

phuong-d-h-nguyen commited on Mar 11

Commit

0dcea97

•

1 Parent(s): 2414d77

Upload folder using huggingface_hub

Files changed (1) hide show

finetuning.py CHANGED Viewed

@@ -6,8 +6,8 @@ import torch
 with open("finetuning_set_v1.0.json") as f:
     dataset = json.load(f)
-torch.cuda.empty_cache()
-torch.cuda.set_per_process_memory_fraction(0.9)
 args = TrainingArguments(
     per_device_train_batch_size=1,

 with open("finetuning_set_v1.0.json") as f:
     dataset = json.load(f)
+# torch.cuda.empty_cache()
+# torch.cuda.set_per_process_memory_fraction(0.9)
 args = TrainingArguments(
     per_device_train_batch_size=1,