networks.py line 808 variable "mean" not defined

NM512 / dreamerv3-torch

Implementation of Dreamer v3 in pytorch.

MIT License

422 stars 96 forks source link

networks.py line 808 variable "mean" not defined #31

Closed lgh0504 closed 1 year ago

lgh0504 commented 1 year ago

    elif self._dist == "normal_1":
        x = self._dist_layer(x)
        dist = torchd.normal.Normal(mean, 1)
        dist = tools.ContDist(torchd.independent.Independent(dist, 1))

     the variable "mean" is not defined

NM512 commented 1 year ago

Thank you for bringing that to my attention. I have fixed the bug.