Skip to content

Multiple workers in pytorch_hyperbolic #8

@bkj

Description

@bkj

I noticed that you can get substantial training speedups by adding eg. num_workers=8 here:
https://github.com/HazyResearch/hyperbolics/blob/master/pytorch/pytorch_hyperbolic.py#L279

I haven't had a chance to test whether this is valid on the other 3 possible DataLoader instantiations.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions