Update models/PDFNet.py
Browse files- models/PDFNet.py +1 -1
models/PDFNet.py
CHANGED
|
@@ -575,7 +575,7 @@ class PDFNet_process(nn.Module):
|
|
| 575 |
|
| 576 |
def build_model(args):
|
| 577 |
if args.back_bone == 'PDFNet_swinB':
|
| 578 |
-
return PDFNet_process(encoder=SwinB(args=args,in_chans=3,pretrained=
|
| 579 |
decoder=PDF_decoder(args=args),depth_decoder=PDF_depth_decoder(args=args),
|
| 580 |
device=args.device, args=args),args.model
|
| 581 |
|
|
|
|
| 575 |
|
| 576 |
def build_model(args):
|
| 577 |
if args.back_bone == 'PDFNet_swinB':
|
| 578 |
+
return PDFNet_process(encoder=SwinB(args=args,in_chans=3,pretrained=False),
|
| 579 |
decoder=PDF_decoder(args=args),depth_decoder=PDF_depth_decoder(args=args),
|
| 580 |
device=args.device, args=args),args.model
|
| 581 |
|