scalping dash also works initially
This commit is contained in:
@ -572,9 +572,17 @@ class EnhancedCNNTrainer:
|
||||
def get_model(self) -> EnhancedCNNModel:
|
||||
"""Get the trained model"""
|
||||
return self.model
|
||||
|
||||
|
||||
def close_tensorboard(self):
|
||||
"""Close TensorBoard writer if it exists"""
|
||||
if hasattr(self, 'writer') and self.writer:
|
||||
try:
|
||||
self.writer.close()
|
||||
except:
|
||||
pass
|
||||
|
||||
def __del__(self):
|
||||
"""Cleanup"""
|
||||
"""Cleanup when object is destroyed"""
|
||||
self.close_tensorboard()
|
||||
|
||||
def main():
|
||||
|
Reference in New Issue
Block a user