search

gemcollector / PIE-G

This is the repo of NeurIPS 2022 paper: "Pre-Trained Image Encoder for Generalizable Visual Reinforcement Learning"
MIT License
12 stars 2 forks source link

BatchNorm update in Pie-g #2

Open sai-prasanna opened 3 months ago

sai-prasanna commented 3 months ago

In the paper, it's mentioned that updating batchnorm is important. I can't find it in the code, maybe I am missing something.

gemcollector commented 3 months ago

Hi there, pie-g will update the statistical params during evaluation: https://github.com/gemcollector/PIE-G/blob/3588cf8b966b7fad5a6c704a09e72ef09bf7afb8/dmcontrol-generalization-benchmark/src/reseval.py#L39. You can check this line.