cuhksz-nlp / R2Gen

165 stars 50 forks source link

Does difference between torch and torchvision versions cause a big difference in evaluation results? #30

Closed Ammexm closed 1 year ago

Ammexm commented 1 year ago

hi~ As I am running the project on RTX3090 cuda 11.6, I have configured the project to run with torch version 1.8.1 and torchvisiond version 0.9.1. However, the difference between the evaluation results and the paper after training is between 1% and 3%. 1.Do I need to follow your requirements to the letter in order to get the original results? Is it possible that different versions of torch and torchvision may cause large differences in the evaluation results? 2.Also, do you use multiple GPUs for your training? If so, can you share the settings?

Looking forward to hearing from you~