world-modelz / dreamax

A scalable Dreamer implementation in JAX
MIT License
11 stars 2 forks source link

Fix `WARNING:root:NaN or Inf found in input tensor.` #19

Open andreaskoepf opened 2 years ago

andreaskoepf commented 2 years ago

Currently during training frequently the warning message WARNING:root:NaN or Inf found in input tensor. appears.