fuqiang

Hello. First of all, I want to thank you for your great work. I'm really interested in your work and i want to fine tune your pre-trained weights on my...

How to convert model to TRT

3

Hello again. as you mentioned in the another issue, i tried to convert your model to TRT with torch2trt. here is a part of my code: dummy_input_to_model_init_encoder = torch.ones((1, 3,...

Hello, I'm working on your awesome project, i just converted pre-trained weights to trt, now i have trouble in creating input_embeddingsFile file, could you please help me to generate that?

training error while setting the batch size more than 1

1

hello, thanks for your great work. i am trying to run your code on my own dataset, but there is an error. when i set batch size to a number...