def test_atari_init(self): """Tests that a DQN agent is initialized.""" train.main([]) shutil.rmtree(FLAGS.base_dir)
import tensorflow as tf from batch_rl.baselines.train import main import dopamine def print_gpus(): ld = tf.config.list_logical_devices() for ldi in ld: print(ldi) if __name__ == '__main__': main()