UltraLight-VM-UNet
UltraLight-VM-UNet copied to clipboard
wurenkai
[arXiv] The official code for "UltraLight VM-UNet: Parallel Vision Mamba Significantly Reduces Parameters for Skin Lesion Segmentation".
请问如何显示tensorboard，train里面的代码没用相关函数～ 我按照vmunet里train.py修改了，但是出现的结果不太对。 我是这么修改的： 1. train.py： - 添加了： global writer writer = SummaryWriter(config.work_dir + 'summary') - 在train_one_epoch添加writer参数： train_one_epoch( train_loader, model, criterion, optimizer, scheduler, epoch, logger, config, writer, scaler=scaler ) 2. engine.py：def...
 Hello author, I downloaded the ph2 dataset from the connection you provided, and its structure is as follows. My question is, do I need to process it before running...
作者您好 我在运行train时发生了以下错误| ../aten/src/ATen/native/cuda/Loss.cu:92: operator(): block: [32,0,0], thread: [32,0,0] Assertion `input_val >= zero && input_val = zero && input_val = zero && input_val = zero && input_val = zero && input_val...
Concern
Parallel processing and feature segmentation methods are well-established in the field of deep learning, particularly in network architectures such as those described in the Inception series. If the primary innovation...
Excellent work. I installed the corresponding packages and attempted to train with a different dataset, resizing the images into 256-sized blocks, and then encountered the following issue. Could you please...
I have some doubts about the flops and params which are computed using thop. Because i think the Mamba layer is not supported there? This would lead zo wrong total...
