keras - 将 Google Colab 设置为使用 Theano 作为后端

Question

我正在关注使用 Theano 作为后端的在线教程。我正在使用默认将 Tensorflow 作为后端的 Google Colab。

我们如何切换到使用 Theano 作为后端？谢谢

标签： kerastheanogoogle-colaboratory

Answer 1

这会有所帮助，

import os
os.environ["KERAS_BACKEND"] = "theano"
import keras.backend
keras.backend.set_image_dim_ordering('th')