diff --git a/configs/mcli/mitchish7.yaml b/configs/mcli/mitchish7.yaml index 6c878c71b..0b0592fbb 100644 --- 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 \