torchrec
torchrec copied to clipboard
pytorch
support serialization of compound module
Summary:
details
- serialization schedule
comp.ebc does NOT require further serialization of its children
comp.ebc.embedding_bags is skipped for further serialization
comp.ebc.embedding_bags.t1 is skipped for further serialization
comp.ebc.embedding_bags.t2 is skipped for further serialization
comp.ebc.embedding_bags.t3 is skipped for further serialization
comp.comp is resumed for serialization
comp.comp Requires further serialization of its children
comp.comp.ebc does NOT require further serialization of its children
comp.comp.ebc.embedding_bags is skipped for further serialization
comp.comp.ebc.embedding_bags.t1 is skipped for further serialization
comp.comp.ebc.embedding_bags.t2 is skipped for further serialization
comp.comp.ebc.embedding_bags.t3 is skipped for further serialization
comp.comp.comp is resumed for serialization
comp.comp.comp Requires further serialization of its children
comp.comp.comp.ebc does NOT require further serialization of its children
comp.comp.comp.ebc.embedding_bags is skipped for further serialization
comp.comp.comp.ebc.embedding_bags.t1 is skipped for further serialization
comp.comp.comp.ebc.embedding_bags.t2 is skipped for further serialization
comp.comp.comp.ebc.embedding_bags.t3 is skipped for further serialization
-
parent_fqn's children
comp.comp.comp {'ebc': EmbeddingBagCollection(
(embedding_bags): ModuleDict(
(t1): EmbeddingBag(10, 4, mode='sum')
(t2): EmbeddingBag(10, 4, mode='sum')
(t3): EmbeddingBag(10, 4, mode='sum')
)
)} None None
comp.comp {'ebc': EmbeddingBagCollection(
(embedding_bags): ModuleDict(
(t1): EmbeddingBag(10, 4, mode='sum')
(t2): EmbeddingBag(10, 4, mode='sum')
(t3): EmbeddingBag(10, 4, mode='sum')
)
), 'comp': CompoundModule(
(ebc): EmbeddingBagCollection(
(embedding_bags): ModuleDict(
(t1): EmbeddingBag(10, 4, mode='sum')
(t2): EmbeddingBag(10, 4, mode='sum')
(t3): EmbeddingBag(10, 4, mode='sum')
)
)
)} None None
comp {'ebc': EmbeddingBagCollection(
(embedding_bags): ModuleDict(
(t1): EmbeddingBag(10, 4, mode='sum')
(t2): EmbeddingBag(10, 4, mode='sum')
(t3): EmbeddingBag(10, 4, mode='sum')
)
), 'comp': CompoundModule(
(ebc): EmbeddingBagCollection(
(embedding_bags): ModuleDict(
(t1): EmbeddingBag(10, 4, mode='sum')
(t2): EmbeddingBag(10, 4, mode='sum')
(t3): EmbeddingBag(10, 4, mode='sum')
)
)
(comp): CompoundModule(
(ebc): EmbeddingBagCollection(
(embedding_bags): ModuleDict(
(t1): EmbeddingBag(10, 4, mode='sum')
(t2): EmbeddingBag(10, 4, mode='sum')
(t3): EmbeddingBag(10, 4, mode='sum')
)
)
)
)} None None
Differential Revision: D58221182
