huawei-noah/SMARTS

Fine tuning pre-trained agent

Aberta

#1.471 aberto em 17 de jun. de 2022

 (1 comentário) (0 reação) (0 responsável)Python (220 forks)auto 404
help wanted

Métricas do repositório

Stars
 (1.129 estrelas)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 30d)

Description

High Level Description Fine-tune the "RL-Agent" provided inside the model zoo, by specifying a new OpenAI Gym environment with customly-defined rewards. We tried several configurations, from understanding how to load the pre-trained weights inside Ray, to instantiate a model and replace the weights with the pre-trained ones, or by trying to registering the model inside Ray, but no strategies leaded us to anything working.

Desired SMARTS version 0.6.1rc1

Operating System Docker, Ubuntu 16.04

Problems The agent is correctly instantiated, but it is not clear how to fine-tune the created pretrained model. We tried to customize the rllib.py by instantiating the agent and passing it to Ray, through tune.run, but we were unsuccessful. Also, inside the rlagent script, there is a warning from developers that the zoo model is not trainable, but it is provided for evaluation only. No further explanations are given.

Guia do colaborador