model = DocFormer_For_IR(config,classes).to(device)
model = DocFormer_For_IR(config,classes).to(device)