File tree Expand file tree Collapse file tree 1 file changed +0
-10
lines changed Expand file tree Collapse file tree 1 file changed +0
-10
lines changed Original file line number Diff line number Diff line change 22
22
from pathlib import Path
23
23
24
24
import jax
25
- import jax .experimental .compilation_cache .compilation_cache as cc
26
25
import jax .numpy as jnp
27
26
import numpy as np
28
27
import optax
@@ -243,12 +242,6 @@ def parse_args():
243
242
action = "store_true" ,
244
243
help = "Whether to dump an initial (before training loop) and a final (at program end) memory profile." ,
245
244
)
246
- parser .add_argument (
247
- "--ccache" ,
248
- type = str ,
249
- default = None ,
250
- help = "Enables compilation cache." ,
251
- )
252
245
parser .add_argument (
253
246
"--tokenizer_name" ,
254
247
type = str ,
@@ -1017,9 +1010,6 @@ def l2(xs):
1017
1010
}
1018
1011
)
1019
1012
1020
- if args .ccache :
1021
- cc .initialize_cache (args .ccache )
1022
-
1023
1013
global_step = step0 = 0
1024
1014
epochs = tqdm (
1025
1015
range (args .num_train_epochs ),
You can’t perform that action at this time.
0 commit comments