about the command of training： ./tools/dist_train.sh /path/to/config 8

Visual-Attention-Network / SegNeXt

Official Pytorch implementations for "SegNeXt: Rethinking Convolutional Attention Design for Semantic Segmentation" (NeurIPS 2022)

Apache License 2.0

796 stars 85 forks source link

about the command of training： ./tools/dist_train.sh /path/to/config 8 #19

Closed Onehundred0906 closed 2 years ago

Onehundred0906 commented 2 years ago

Hello, very appreciate to give us so excellent article to solve visual tasks, can you explain the path parameter? I cant't find some information about it in code, very grateful！ segnext_config

uyzhang commented 2 years ago

For example: ./tools/dist_train.sh local_configs/segnext/tiny/segnext.tiny.512x512.ade.160k.py 8

Onehundred0906 commented 2 years ago

Thanks, i have another question, my computer has only a gpu, your test command is normal to use or not ? i change the command :./tools/dist_test.sh /path/to/config /path/to/checkpoint_file 1 --eval mIoU. only "8" is instead of "1". I run it but occur some mistakes.

uyzhang commented 2 years ago

Because a single GPU cannot use dist_test.sh