Sayuri icon indicating copy to clipboard operation
Sayuri copied to clipboard

怎么使用自己定义的模型结构训练呢?

Open Nightbringers opened this issue 1 year ago • 1 comments

不会c++, 会torch, 想试一下把模型换成自己的结构训练一下试试,该怎么做呢,可以给个例子吗

Nightbringers avatar Feb 15 '24 11:02 Nightbringers

這樣的話可能沒有辦法,引擎端的神經網路是使用 CUDA 和 C++ 撰寫的,改變網路結構需要重寫之

CGLemon avatar Feb 15 '24 16:02 CGLemon