Added a basic torch profiler that can be used in config during development to find some obvious issues.

This commit is contained in:
Jaret Burkett
2025-06-17 13:03:39 -06:00
parent ff617fdaea
commit 989ebfaa11
4 changed files with 53 additions and 66 deletions

View File

@@ -15,7 +15,6 @@ class BaseJob:
self.config = config['config']
self.raw_config = config
self.job = config['job']
self.torch_profiler = self.get_conf('torch_profiler', False)
self.name = self.get_conf('name', required=True)
if 'meta' in config:
self.meta = config['meta']