Skip to content

Commit 87a3f95

Browse files
715 ml memory (#865)
* #715 Support for no ml memory limit * API update
1 parent c695989 commit 87a3f95

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

autosklearn/automl.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -304,7 +304,11 @@ def _do_dummy_prediction(self, datamanager, num_run):
304304
return num_run
305305

306306
self._logger.info("Starting to create dummy predictions.")
307-
memory_limit = int(self._ml_memory_limit)
307+
308+
memory_limit = self._ml_memory_limit
309+
if memory_limit is not None:
310+
memory_limit = int(memory_limit)
311+
308312
scenario_mock = unittest.mock.Mock()
309313
scenario_mock.wallclock_limit = self._time_for_task
310314
# This stats object is a hack - maybe the SMAC stats object should

autosklearn/estimators.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,7 @@ def __init__(
9797
Memory limit in MB for the machine learning algorithm.
9898
`auto-sklearn` will stop fitting the machine learning algorithm if
9999
it tries to allocate more than `ml_memory_limit` MB.
100+
If None is provided, no memory limit is set.
100101
101102
include_estimators : list, optional (None)
102103
If None, all possible estimators are used. Otherwise specifies

0 commit comments

Comments
 (0)