Criação de "ENV" próprio

LisandraMoura / Mario-kart-RL

0 stars 0 forks source link

Criação de "ENV" próprio #3

Open mendesLet opened 5 days ago

mendesLet commented 5 days ago

Agora que conseguimos rodar um modelo precisamos adaptar para nossa situação

[x] Estudar repositório para converter o treinamento para GPU
[ ] Estudar repositório para tentar aplicar o treinamento em outros levels
[ ] Criar função que salva checkpoints a cada 500.000 steps

A ideia é adaptar o repositório mario-kart-agent de uma maneira que propomos para o projeto e eventualmente iremos trazer para nosso repositório

EdwardSJ151 commented 1 day ago

Como as redes são pequenas, não vale a pena usar GPU a não ser que nossa rede fique mais parruda. Porém, isso é só para os calculos. Não tem como paralelizar basicamente nada pois a gente precisa de um worker por ambiente, já que os ambientes são sequenciais.