search

CSAILVision / semantic-segmentation-pytorch

Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset
http://sceneparsing.csail.mit.edu/
BSD 3-Clause "New" or "Revised" License
4.96k stars 1.1k forks source link

training stopped.. #213

Closed MyoungHaSong closed 4 years ago

MyoungHaSong commented 4 years ago

I run this code. python3 train.py --gpus 0,1 --cfg config/ade20k-resnet101-upernet.yaml

but training was stopped..

image

my environment is nvidia-docker + titan XP * 2 , CPU =i9-9900K what should I do ?

hangzhaomit commented 4 years ago

Should be a GPU memory issue. Try adjust batch size.