Skip to content

Ask about train setting #6

Description

@WISHF-Amoy

Hello, thanks for your great work! I want to re-train this work in whole BEAT2 25 speakers, and found that the description of training setting of LLM in paper is different with README. In paper, it said the batch sizes per GPU are 32, 20, and 12, respectively. The gradient accumulation steps are set to 4, 6, and 10, , but in README, it said num_gpus × batch_size × gradient_accumulation_steps ≈ 256, and in three stages' config.yaml, default batch size is both 128 for 2 GPUs, and gradient_accumulation_steps is also same as default value 1.

So may I ask which setting is actually yours, which make it possible for me to making a fair compare, thanks a lot.

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