Update models/PDFNet.py
Browse files- models/PDFNet.py +1 -1
models/PDFNet.py
CHANGED
@@ -506,6 +506,6 @@ class PDFNet_process(nn.Module):
|
|
506 |
|
507 |
def build_model(args):
|
508 |
if args.back_bone == 'PDFNet_swinB':
|
509 |
-
return PDFNet_process(encoder=SwinB(args=args,in_chans=3,pretrained=
|
510 |
decoder=PDF_decoder(args=args),depth_decoder=PDF_depth_decoder(args=args),
|
511 |
device=args.device, args=args),args.model
|
|
|
506 |
|
507 |
def build_model(args):
|
508 |
if args.back_bone == 'PDFNet_swinB':
|
509 |
+
return PDFNet_process(encoder=SwinB(args=args,in_chans=3,pretrained=False),
|
510 |
decoder=PDF_decoder(args=args),depth_decoder=PDF_depth_decoder(args=args),
|
511 |
device=args.device, args=args),args.model
|