|
@@ -43,13 +43,15 @@ class FineTuneParser(GPUParser):
|
|
|
|
|
|
|
|
def __del__(self):
|
|
def __del__(self):
|
|
|
try:
|
|
try:
|
|
|
- self._file_handler.flush()
|
|
|
|
|
|
|
+ if getattr(self, "_file_handler", None) is not None:
|
|
|
|
|
+ self._file_handler.flush()
|
|
|
except Exception as e:
|
|
except Exception as e:
|
|
|
warnings.warn("Could not flush logs to file: {}".format(e))
|
|
warnings.warn("Could not flush logs to file: {}".format(e))
|
|
|
|
|
|
|
|
|
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
def __init__(self, *args, **kwargs):
|
|
|
super(FineTuneParser, self).__init__(*args, **kwargs)
|
|
super(FineTuneParser, self).__init__(*args, **kwargs)
|
|
|
|
|
+ self._file_handler = None
|
|
|
|
|
|
|
|
add_dataset_args(self)
|
|
add_dataset_args(self)
|
|
|
add_model_args(self)
|
|
add_model_args(self)
|