fix TrainSpec in simple_fsdp experiments

yiming0416 · yiming0416 · commit 0d75b042c8b7 · 2025-10-10T16:30:32.000-07:00
diff --git a/torchtitan/experiments/simple_fsdp/deepseek_v3/__init__.py b/torchtitan/experiments/simple_fsdp/deepseek_v3/__init__.py
@@ -21,7 +21,6 @@
 
 def get_train_spec() -> TrainSpec:
     return TrainSpec(
-        name="simple_fsdp.deepseek_v3",
         model_cls=SimpleFSDPDeepSeekV3Model,
         model_args=deepseekv3_configs,
         parallelize_fn=parallelize_deepseekv3,
diff --git a/torchtitan/experiments/simple_fsdp/llama3/__init__.py b/torchtitan/experiments/simple_fsdp/llama3/__init__.py
@@ -20,7 +20,6 @@
 
 def get_train_spec() -> TrainSpec:
     return TrainSpec(
-        name="simple_fsdp.llama3",
         model_cls=SimpleFSDPTransformer,
         model_args=llama3_configs,
         parallelize_fn=parallelize_llama,
