Keras
切換後端
切換後端
import os
#"jax", "tensorflow", "torch"
os.environ["KERAS_BACKEND"] = "torch"
import keras
需要在import keras 之前,設定好後端框架種類
import os
#"jax", "tensorflow", "torch"
os.environ["KERAS_BACKEND"] = "torch"
import keras
需要在import keras 之前,設定好後端框架種類