Skip to content

DFormerv2复现问题请教 #51

@hudfdfdf

Description

@hudfdfdf
  您好,首先感谢您的代码公开分享。我正在学习复现您的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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions