klein

[Feature] specify gpus in pipeline

2

### Motivation How to specify gpus id in pipeline? I need to run two models simultaneously ### Related resources _No response_ ### Additional context _No response_

Does training is stable?

2

Hi, authors: Thanks for your work and code. I have tried to run your code on 2 A100. But the result is ~7, which seems hard to achieve 5.26 on...

The loss detail saved in vis_data is a jsonl file instead of a json file.

Congratulations to your work. Are all the training data the one thousand examples provided?

Thanks for your contribution and we appreciate it a lot. The following instructions would make your pull request more healthy and more easily get feedback. If you do not understand...