Skip to content

【25-Q4-生态建设】模型迁移-研发效能部-模型训练-在PyTorch框架上支持segformer在 cityscapes上的训练#460

Open
0217wyt wants to merge 1 commit intoTecorigin:mainfrom
0217wyt:segformer
Open

【25-Q4-生态建设】模型迁移-研发效能部-模型训练-在PyTorch框架上支持segformer在 cityscapes上的训练#460
0217wyt wants to merge 1 commit intoTecorigin:mainfrom
0217wyt:segformer

Conversation

@0217wyt
Copy link
Contributor

@0217wyt 0217wyt commented Jan 4, 2026

● 当前软件栈版本:
image

● 源码参考链接:https://github.com/open-mmlab/mmsegmentation/blob/main/README_zh-CN.md
● commit id:0217wyt@ https://github.com/0217wyt
● 工作目录:PyTorch/build-in/Segmentation/segformer
● 训练内容:使用1张TECO_AICARD_01芯片,在PyTorch框架上支持 segformer 在 cityscapes 数据集上的训练。
● 运行脚本如下:
bash tools/dist_train.sh configs/segformer/segformer_mit-b0_8xb1-160k_cityscapes-1024x1024.py 8

● 100iters损失:
image

image

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants