問題導(dǎo)出

為了更方便地實現(xiàn)自己的一些想法，我最終還是向OpenAI的baselines屈服了，因為自己的一些實現(xiàn)總是難與各大論文給出的結(jié)果進行對比。于是我今天安裝了baselines，其中還是有一些坑的，這里主要介紹在安裝完成之后，運行示例時cartpole.py所遇到的坑，它叫做ImportError。

ImportError

我們先cd到文件夾baselines/baselines/deepq/experiments下，然后運行下面語句：

$ python3 train_cartpole.py

然后就報錯了……

從圖中我們可以看出，在文件train_cartpole.py中有如下語句：

from baselines import deepq

而在我們導(dǎo)入deepq的時候，將會執(zhí)行deepq文件夾下的__init__.py文件，于是我們打開這一文件：

from baselines.deepq import models  # noqa
from baselines.deepq.build_graph import build_act, build_train  # noqa
from baselines.deepq.simple import learn, load  # noqa
from baselines.deepq.replay_buffer import ReplayBuffer, PrioritizedReplayBuffer  # noqa
def wrap_atari_dqn(env):
    from baselines.common.atari_wrappers import wrap_deepmind
    return wrap_deepmind(env, fram_stack=True, scale=True)

在報錯中，提到了上面的這一語句：

from baselines.deepq.simple import learn, load  # noqa

同時，報錯中提到，在simple.py中有：

from baselines import deepq

問題漸漸明朗了，在我們import deepq的時候，將要執(zhí)行__init__.py文件，這個文件將會從simple.py中把learn和load導(dǎo)入到deepq的命名空間下。而simple.py又想要import deepq，這也就構(gòu)成了所謂的“Looping Import”。那我們應(yīng)該如何解決呢？

我們先將simple.py中的下面這條語句注釋掉：

from baselines import deepq

然后運行：

$ python3 train_cartpole.py

發(fā)現(xiàn)又出錯了：

不過這個錯誤是肯定會出的，我們無緣無故注釋掉了人家寫的一個import語句，能不錯嘛。不過不要慌，我們繼續(xù)分析，為什么這里要import deepq呢？就是為了執(zhí)行下面的語句：

act = deepq.build_act(**act_params)
...
act, train, update_target, debug = deepq.build_train(...)

這里因為我們在__init__.py中將build_act和build_train方法導(dǎo)入到了deepq的命名空間下，所以這里可以直接在deepq的命名空間中去調(diào)用build_act和build_action。

如果我們要在simple.py文件中使用deepq.build_act的話，那么必然要在simple.py中導(dǎo)入deepq，而且也需要在__init__.py中導(dǎo)入simple，這么一來，就會出現(xiàn)“Looping Import”問題了。

解決方案

所以我們選擇的解決方案是：

a) 將__init__.py中的下列語句注釋掉：

from baselines.deepq.build_graph import build_act, build_train # noqa

b) 將simple.py中的下列語句換掉：

from baselines import deepq

換為：

from baselines.deepq.build_graph import build_act, build_train

c) 修改下列語句：

act = deepq.build_act(**act_params)
...
act, train, update_target, debug = deepq.build_train(...)

改為：

act = build_act(**act_params)
...
act, train, update_target, debug = build_train(...)

d) 運行：

$ python3 train_cartpole.py

OK，問題解決。

對了，在OpenAI Baselines里面，有一些代碼用到了Python3.6的特性，比如直接(None, *tuple)，這在Python3.4是不允許的，所以建議大家直接安裝conda環(huán)境，并裝好Python3.6。

以上就是baselines示例程序train_cartpole.py的ImportError的詳細(xì)內(nèi)容，更多關(guān)于train_cartpole.py的ImportError的資料請關(guān)注腳本之家其它相關(guān)文章！

最新評論