agentscope-ai/Trinity-RFT

Reward function中的Reward Model应该在哪里初始化？

Open

#312 aberto em 10 de out. de 2025

Ver no GitHub
 (2 comments) (0 reactions) (0 assignees)Python (70 forks)github user discovery
good first issue

Métricas do repositório

Stars
 (648 stars)
Métricas de merge de PR
 (Métricas PR pendentes)

Description

Reward function中的Reward Model在哪里初始化是最好的呢；我目前是在workflow类下初始化的，但是它只能加载到cpu，然后推理打分时会非常非常慢，以至于超时报错

Guia do colaborador

Direção de pesquisa
Investigue como mover a inicialização do Reward Model para a GPU. Verifique as utilitárias de carregamento de modelo do framework e certifique se de que o dispositivo está definido como CUDA, se disponível. Procure exemplos de inicialização em GPU em outras partes do código.
Pilha de tecnologia
python
Domain
aimachine learning
Tipo Issue
Performance
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
2
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-3 horas
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
PythonPyTorch
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
70