File size: 449 Bytes
3cd1076 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | CKPT_PATH="YOUR_CHECKPOINT_PATH"
STEPS=1024
python -u -m main \
mode=sample_eval \
seed=1 \
model=small \
model.length=128 \
data=lm1b-wrap \
algo=flm \
eval.checkpoint_path=$CKPT_PATH \
loader.batch_size=2 \
loader.eval_batch_size=32 \
sampling.num_sample_batches=1 \
sampling.steps=$STEPS \
algo.double_temb=False \
eval.disable_ema=False \
+wandb.offline=true \ |