問題タブ [xla]
For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.
61 問題
0
投票する
0
に答える
117
参照
pytorch - モデルの作成中に Colab TPU がクラッシュして再起動する
私は Pytorch を使用しており、Colab GPU でモデルをトレーニングできます。私のモデルはそれほど大きくありません。約10Mのパラメータです。Colab TPU でトレーニングしようとすると、セッションがクラッシュして突然再起動します。XLA のナイトリー バージョンとデフォルト バージョンの両方を試しました。RAMが増加しないことを確認しました。
ご協力いただきありがとうございます
0
投票する
0
に答える
31
参照
tensorflow - tensorflow で粗粒度の op レベル グラフを取得する方法
tensorflow を使用して、完全な計算グラフ (前方、後方、およびパラメーターの更新を含む) を取得したいと考えています。私は tf.functions を試しましたが、得られたグラフは細かすぎます。多くの演算 (Adam など) が小さな演算子 (add、mul、div など) に分割されているためです。粗粒度の操作レベルのグラフを取得できる方法はありますか? どうもありがとう！