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:
|
||||
return
|
||||
|
||||
session_config.isolate_session_state = True
|
||||
|
||||
assert self._task_type
|
||||
assert self._task_id is not None
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user