autodeeplab icon indicating copy to clipboard operation
autodeeplab copied to clipboard

fix channel mismatch in new_model.py when i==1

Open JingweiZhang12 opened this issue 4 years ago • 0 comments

  1. discard useless parameter 'criterion' in Class newModel
  2. fix channel mismatch in new_model.py when i==1
  3. change the default cityscapes epoch to 90
  4. remove args.gpu_ids, replace the 'len(agrs.gpu_ids)' by 'torch.cuda.device_count()'
  5. some parameters changes considering practical training, not matter

JingweiZhang12 avatar Sep 17 '19 02:09 JingweiZhang12