Closed ydhongHIT closed 2 years ago
Your labels do not range from 0 to num_classes.
/home/ydhong/.conda/envs/detectron/lib/python3.6/site-packages/torch/optim/lr_scheduler.py:100: UserWarning: Detected call of lr_scheduler.step()
before optimizer.step()
. In PyTorch 1.1.0 and later, you should call them in the opposite order: optimizer.step()
before lr_scheduler.step()
. Failure to do this will result in PyTorch skipping the first value of the learning rate schedule.See more details at https://pytorch.org/docs/stable/optim.html#how-to-adjust-learning-rate
"https://pytorch.org/docs/stable/optim.html#how-to-adjust-learning-rate", UserWarning)
/opt/conda/conda-bld/pytorch_1573049301898/work/aten/src/THCUNN/SpatialClassNLLCriterion.cu:104: void cunn_SpatialClassNLLCriterion_updateOutput_kernel(T , T , T , long , T *, int, int, int, int, int, long) [with T = float, AccumT = float]: block: [32,0,0], thread: [544,0,0] Assertion t >= 0 && t < n_classes
failed.
Traceback (most recent call last):
File "train.py", line 317, in
Do you use this codes with CamVid? There are some bugs.