Hugging Faceモデル上のカスタムモデルで作成

[Kumar9 (Kumar C)]

PyTorchで独自のモデルを作成し、モデルにホストしようとしていますが、うまくいきません。何かアイデアはありますか？
モデルのトレーニングにはColabを使用しました。