ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Hot fix, release

Browse Source
This commit is contained in:
Allegro AI 2019-06-30 10:53:38 +03:00 committed by GitHub
parent 4174d9345e
commit a6e70f783a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
trains/task.py
View File

@ -33,7 +33,7 @@ from .utilities.args import argparser_parseargs_called, get_argparser_last_args,
from .binding.frameworks.pytorch_bind import PatchPyTorchModelIO from .binding.frameworks.pytorch_bind import PatchPyTorchModelIO
from .binding.frameworks.tensorflow_bind import PatchSummaryToEventTransformer, PatchTensorFlowEager, \ from .binding.frameworks.tensorflow_bind import PatchSummaryToEventTransformer, PatchTensorFlowEager, \
PatchKerasModelIO, PatchTensorflowModelIO PatchKerasModelIO, PatchTensorflowModelIO
from .utilities.resource_monitor import ResourceMonitor # from .utilities.resource_monitor import ResourceMonitor
from .binding.matplotlib_bind import PatchedMatplotlib from .binding.matplotlib_bind import PatchedMatplotlib
from .utilities.seed import make_deterministic from .utilities.seed import make_deterministic
@ -226,8 +226,8 @@ class Task(_Task):
PatchTensorflowModelIO.update_current_task(task) PatchTensorflowModelIO.update_current_task(task)
PatchPyTorchModelIO.update_current_task(task) PatchPyTorchModelIO.update_current_task(task)
if auto_resource_monitoring: if auto_resource_monitoring:
task._resource_monitor = ResourceMonitor(task) # task._resource_monitor = ResourceMonitor(task)
task._resource_monitor.start() pass # task._resource_monitor.start()
# Check if parse args already called. If so, sync task parameters with parser # Check if parse args already called. If so, sync task parameters with parser
if argparser_parseargs_called(): if argparser_parseargs_called():
parser, parsed_args = get_argparser_last_args() parser, parsed_args = get_argparser_last_args()