Enable compile

allenai · Apr 3, 2024 · d49a91d · d49a91d
1 parent 0ef892b
commit d49a91d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/configs/mcli/mitchish7.yaml b/configs/mcli/mitchish7.yaml
@@ -58,7 +58,8 @@ command: |-
     --fsdp.wrapping_strategy=by_block_and_size \
     --fsdp.sharding_strategy=SHARD_GRAD_OP \
     --save_folder=runs/ \
-    --activation_checkpointing=fine_grained \
+    --activation_checkpointing=null \
+    --compile.fullgraph=false \
     --fused_loss=true \
     --device_train_microbatch_size=2 \
     --global_train_batch_size=1024 \