the code is running without any results and output about dreamer HOT 5 CLOSED

while step < config.steps:
print('Start evaluation.')
tools.simulate(
functools.partial(agent, training=False), test_envs, episodes=1)

it seems that the code is running here, and it can't stop and output anything.

from dreamer.

It seems like it's running as intended. What are you expecting to see?

from dreamer.

I want to debug the following code, but it seems that there will be a long time to finish this line:
while step < config.steps:
print('Start evaluation.')
tools.simulate(functools.partial(agent, training=False), test_envs, episodes=1)

So can I know how long it will take? Thanks.

from dreamer.

I think I have seen it before. Check if your GPU is correctly used (with some Volatile GPU-Util).

from dreamer.

Right, maybe you're not using a GPU or your environment is very slow. If you switch to DreamerV2, you can also verify that the code is running correctly by using --configs defaults atari debug. The debug config will reduce the batch size and increase the logging frequency.

from dreamer.