P-tuning

P-tuning copied to clipboard

就是代码能跑起来的python，tensorflow，cuda，cudnn的版本

hitstu0

转pb文件，PtuningEmbedding层报错

2

https://github.com/bojone/P-tuning/blob/aec82943f21268a6c813877ac055631e57cb96c3/bert.py#L122 苏神，再用P-tuning代码保存为h5格式后，转pb文件时（参考https://github.com/bojone/bert4keras/issues/194），会报错ValueError: Unknown layer: PtuningEmbedding，麻烦问下知道这是为啥吗？代码如下 ''' import os os.environ['TF_KERAS'] = '1' import numpy as np import pandas as pd from bert4keras.backend import keras,K from bert4keras.layers import Loss, Embedding from bert4keras.tokenizers...

MonkeyTB