cebain comments

Repositories
Issues
Comments

Results 3 comments of


                                            cebain

for

well，for data without labels, if you want to turn them into h5, you can simple write picture data as 'image', and for 'label' you can also write picture data or...

selective_scan_cuda

how can I turn the checkpoint .ckpt into pt, so i can use it by libtorch, while selective_scan_cuda is a CUDA C++ extension, which can not be recognized by torch.jit.scritpt?

deployin C++ using libtorch or TensorRT.

when I try to use torch.jit.script to turn .ckpt into .pt, the following error occured: RuntimeError: Python builtin is currently not supported in Torchscript: File "/home/adminroot/Documents/ccc/pycharmProjects/UNetMamba-main/unetmamba_model/classification/models/vmamba.py", line 286 def SelectiveScanCore(u,...