huawei-noah/SMARTS

Fine tuning pre-trained agent

开放

#1,471 创建于 2022年6月17日

 (1 条评论) (0 个反应) (0 位负责人)Python (220 个派生)auto 404
help wanted

仓库指标

星标
 (1,129 个星标)
PR 合并指标
 (PR 指标待抓取)

描述

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.

贡献者指南