Allen Liu

Choose GPU to use?

2

Hi, could I add some code at the beginning of train.py to choose which GPU to use? Like set the os.environ.

network.py

3

static method should not have "self" maybe