Skip to content

why it has to use share_memory_ #262

@liang00fan

Description

@liang00fan

I've set config.input_threads to 1 and removed self.potentials[i].share_memory_() from my code. otherwise it will report memory issue. But self.potentials will be changed for no reason, I don't know why . it would work when share_memory_ is added. why?

for i, _ in enumerate(self.pot_trees):
self.potentials[i].share_memory_()

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions