DeepPros/DeepDR-LLM

Missing Requirements and initial errors

Opened this issue · 0 comments

The following need to be added to the requirements.txt
scikit-learn
einops
timm
torchvision

After that and downloading the ViT weights per instructions the following error is thrown (weights were added to line 37 of test.py)

`Traceback (most recent call last):
  File "/home/robert/china_dr/DeepDR-LLM/Module2/test.py", line 57, in <module>
    DR_model.load_state_dict(torch.load(Weights["DR"][0]), strict=True)
  File "/home/robert/china_dr/.test/lib/python3.9/site-packages/torch/nn/modules/module.py", line 2215, in load_state_dict
    raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for VisionTransformer:

This seems to indicate model is loaded before it is defined in code. Will advise as I troubleshoot more.