Add more options to profilers (#20)

IntelLabs · Dec 12, 2022 · 6184ca7 · 6184ca7
1 parent 5eed680
commit 6184ca7
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/mart/configs/debug/profiler.yaml b/mart/configs/debug/profiler.yaml
@@ -7,6 +7,9 @@ defaults:
 
 trainer:
   max_epochs: 1
-  profiler: "simple"
-  # profiler: "advanced"
-  # profiler: "pytorch"
+  profiler:
+    _target_: pytorch_lightning.profiler.SimpleProfiler
+    # _target_: pytorch_lightning.profiler.AdvancedProfiler
+    # _target_: pytorch_lightning.profiler.PyTorchProfiler
+    dirpath: ${paths.output_dir}
+    filename: profiler_log