cebain

train own 2D pictures

2

hello，thanks for your revision very much。 I still want to ask a question。 We can see that the h5 file includes both image and label data， and when we train，for...

for

3

hello，thanks for your revision very much。 I still want to ask a question。 We can see that the h5 file includes both image and label data， and when we train，for...

Has anyone successfully deploy Mamba by libtorch? when I use torch.jit.script to turn ckpt into .pt, there is a fault Python builtin is currently not supported in Torchscript: File "/home/adminroot/Documents/ccc/pycharmProjects/UNetMamba-main/unetmamba_model/classification/models/vmamba.py",...

你好，想问下模型有通过libtorch 或者tensorRT部署过吗

deployin C++ using libtorch or TensorRT.

2

I'd like to inquire whether this code has been successfully deployed in C++ using libtorch or TensorRT. I've encountered numerous issues when attempting to deploy VMamba, and I'm wondering if...

看到代码中有cuda 底层编程，想请问下该模型是否可以导出为onnx呢

when I try to export the trained .pth to onnx, it keeps raising error like below, but I can't find any use of adaptive_avg_pool2d in the model, how can i...