Tennineee commited on
Commit
531ca1b
·
verified ·
1 Parent(s): 447dfce

Update models/PDFNet.py

Browse files
Files changed (1) hide show
  1. 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=True),
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