pytorch2keras - PyPI
https://pypi.org/project/pytorch2keras14.05.2020 · from pytorch2keras.converter import pytorch_to_keras # we should specify shape of the input tensor k_model = pytorch_to_keras (model, input_var, [(10, None, None,)], verbose = True) That's all! If all the modules have converted properly, the Keras model will be stored in the k_model variable. API. Here is the only method pytorch_to_keras from ...