2021-06-01から1ヶ月間の記事一覧

TPUの重み保存

qiita.com def get_model_weights_as_numpy(model): weights = {} for v in model.weights: # model.weightsで各Layerの重みを取り出し # 各variableはnumpyメソッドでnumpy配列に変換できる weights[v.name] = v.numpy() return {'model_name': model.name,…