I have been trying to use the “Train FCN on Pascal VOC Dataset” train.py for a while and cannot figure out how to fix this error. I am trying to train using a single gpu.
My understanding is that I have to initialize the model on gpu but I am not sure how:
model = get_segmentation_model(model=args.model, dataset=args.dataset, backbone=args.backbone, norm_layer=args.norm_layer, norm_kwargs=args.norm_kwargs, aux=args.aux) model.cast(args.dtype) print(model) self.net = DataParallelModel(model, args.ctx, args.syncbn) self.evaluator = DataParallelModel(SegEvalModel(model), args.ctx)
I would be thankful for advise