Added a basic torch profiler that can be used in config during development to find some obvious issues.
This commit is contained in:
@@ -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']
|
||||
|
||||
Reference in New Issue
Block a user