fffiloni commited on
Commit
666f4a3
·
verified ·
1 Parent(s): 2fadfd7

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -0
main.py CHANGED
@@ -4,6 +4,7 @@ import os
4
 
5
  import blobfile as bf
6
  import torch
 
7
  from datasets import load_dataset
8
  from pytorch_lightning import seed_everything
9
  from tqdm import tqdm
 
4
 
5
  import blobfile as bf
6
  import torch
7
+ import gc
8
  from datasets import load_dataset
9
  from pytorch_lightning import seed_everything
10
  from tqdm import tqdm