Skip to content

How to train on 4 3090 GPUs? #124

@atti0127

Description

@atti0127

For 'Restormer/train.sh', I simply changed --nproc_per_node to 4

In case of 'RealDenoising_Restormer.yml', how should I change? Can simply change num_gpu to 4, but have no idea how to handle 'batch_size_per_gpu' and 'mini_batch_sizes'. Default is 8 and [8,5,4,2,1,1], respectively.

Could anyone share the experimental results using 4 gpus?

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