Skip to content

Batch_size setting problem #3626

@xinliangSun

Description

@xinliangSun

[Batch_size]

When calling the ContrastiveDataLoader to split the data, the batch_size is passed via self.data_loader_kwargs. However, when the parent class ConcatDataLoader(DataLoader) is invoked, batch_size is defined again with a fixed value. This causes a problem: the batch size configured for the model cannot be changed by the external parameter.

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions