DFormer icon indicating copy to clipboard operation
DFormer copied to clipboard

DFormerv2复现问题请教

Open hudfdfdf opened this issue 3 months ago • 0 comments

  您好,首先感谢您的代码公开分享。我正在学习复现您的DFormerv2,源码训练NYUDepthv2数据集没有问题,但是在源码训练SUNRGBD数据集的时候,没有能够复现论文中的mIoU。和您源码的区别是,我使用了1张80G显存的A100显卡训练。附件是训练DFormerv2_S训练的LOG,希望可以指导一下,训练方式是否有问题,谢谢!

GPUS=1 NNODES=1 NODE_RANK=${NODE_RANK:-0} PORT=${PORT:-29122} MASTER_ADDR=${MASTER_ADDR:-"127.0.0.1"}

export CUDA_VISIBLE_DEVICES="0" export TORCHDYNAMO_VERBOSE=1

PYTHONPATH="$(dirname $0)/..":"$(dirname $0)":$PYTHONPATH
torchrun
--nnodes=$NNODES
--node_rank=$NODE_RANK
--master_addr=$MASTER_ADDR
--nproc_per_node=$GPUS
--master_port=$PORT
utils/train.py
--config=local_configs.SUNRGBD.DFormerv2_S --gpus=$GPUS
--no-sliding
--no-compile
--syncbn
--mst
--compile_mode="default"
--no-amp
--val_amp
--pad_SUNRGBD
--no-use_seed \

log_2025_10_11_17_30_58.log

hudfdfdf avatar Oct 22 '25 06:10 hudfdfdf