Missing Requirements and initial errors
Opened this issue · 0 comments
aeye-slater commented
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.