Update process.py

Author: wbw520

process.py

@@ -16,7 +16,7 @@ def main():
     model.to(device)
     checkpoint = torch.load(os.path.join(args.output_dir,
             f"{args.dataset}_{args.base_model}_cls{args.num_classes}_cpt{args.num_cpt}_" +
-    f"{'use_slot_' + args.cpt_activation if not args.pre_train else 'no_slot'}3.pt"), map_location="cuda:0")
+    f"{'use_slot_' + args.cpt_activation if not args.pre_train else 'no_slot'}.pt"), map_location="cuda:0")
     model.load_state_dict(checkpoint, strict=True)
     model.eval()