Comment out profiler by default. Gotta figure out how to use the conf…

…ig files.
PrimeIntellect-ai · Jan 10, 2025 · b6083f9 · b6083f9
1 parent f79e9ea
commit b6083f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/zeroband/train.py b/src/zeroband/train.py
@@ -472,7 +472,7 @@ def train(config: Config):
     torch.cuda.set_device(world_info.local_rank)
 
     config = Config(**parse_argv())
-    config.train.memory_profiler = MemoryProfilerConfig(snapshot_dir="logs/", freq=1)
+    # config.train.memory_profiler = MemoryProfilerConfig(snapshot_dir="logs/", freq=1)
     logger.debug(f"config: {config.model_dump()}")
 
     try: