mirror of
https://github.com/pytorch/pytorch.git
synced 2025-11-05 08:24:57 +08:00
Temporarily disables the checkpoints for the readers.
Summary: The hive reader checkpoints are broken because of D5582328. This breaks our offline simulator test as well. This is a temporary fix that disables the checkpoints for readers. Reviewed By: azzolini Differential Revision: D5637719 fbshipit-source-id: 4f31ae534cb7e981fcacbb721cbb2420249fad91
This commit is contained in:
committed by
Facebook Github Bot
parent
1db7a99249
commit
49ec942825
@ -235,6 +235,12 @@ class MultiNodeCheckpointManager(object):
|
||||
assert self._node_managers is not None, 'init must be called first.'
|
||||
with TaskGroup(WorkspaceType.GLOBAL) as task_group:
|
||||
for node, manager in self._node_managers:
|
||||
# TODO(aartibasant, T21070353): Enable the checkpoints for
|
||||
# readers.
|
||||
# The checkpointing for readers is broken because of D5582328.
|
||||
# Disabling the reader checkpoints until it is fixed.
|
||||
if "reader" in str(node):
|
||||
continue
|
||||
with Node(node):
|
||||
func(manager, *args, **kw)
|
||||
return task_group
|
||||
|
||||
Reference in New Issue
Block a user