Temporarily remove isolate_session_state in CollectiveAllReduceStrategy.
PiperOrigin-RevId: 214119090
This commit is contained in:
parent
e692dda4c8
commit
adea2433eb
@ -235,8 +235,6 @@ class CollectiveAllReduceStrategy(mirrored_strategy.MirroredStrategy):
|
|||||||
if not session_config or not self._cluster_spec:
|
if not session_config or not self._cluster_spec:
|
||||||
return
|
return
|
||||||
|
|
||||||
session_config.isolate_session_state = True
|
|
||||||
|
|
||||||
assert self._task_type
|
assert self._task_type
|
||||||
assert self._task_id is not None
|
assert self._task_id is not None
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user