imai-laboratory / nec

Neural Episodic Control implementation in TensorFlow
9 stars 0 forks source link

Add evaluator #12

Closed takuseno closed 6 years ago

takuseno commented 6 years ago

Add evaluator

$ python train.py [--record] [--eval-render]

caution

To record rendered game images in Box environment, gym requires window to be open, which causes crash with SSH over connection. To handle it, add the following codes.

from pyvirtualdisplay import Display

# virtual display settings to render gym
v_display = Display(visible=0, size=(1400, 900))
v_display.start()

@smatsumori