Tennineee commited on
Commit
80e155d
·
verified ·
1 Parent(s): c7202d2

Update models/PDFNet.py

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