Skip to content

accelerate support for train_finetune.py with one GPU #98

@yl4579

Description

@yl4579

Request per #96 (comment). Current finetuning script is quite slow because it doesn't support accelerate due to #7, but a lot of people are finetuning using a single GPU, so we don't really have to worry about DP or DDP and can directly turn to accelerate which supports mixed precision that saves time and RAM. It would be great if someone could implement this feature for a single GPU. For multi-GPU we are still trying to solve #7 .

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or requesthelp wantedExtra attention is needed

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions