Python mainの例

コード例 #1

ファイル: atari_init_test.py プロジェクト: alhamzah/batch_rl

 def test_atari_init(self):
     """Tests that a DQN agent is initialized."""
     train.main([])
     shutil.rmtree(FLAGS.base_dir)

コード例 #2

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()