RuntimeError: permute(sparse_coo): number of dimensions in the tensor input does not match the length of the desired ordering of dimensions i.e. input.dim() = 4 is not equal to len(dims) = 2

[zpylyuuuu]

Traceback (most recent call last):
File "D:\few-shot-code\2023CaFo-main\CaFo-main\main.py", line 276, in
main()
File "D:\few-shot-code\2023CaFo-main\CaFo-main\main.py", line 234, in main
dino_cache_keys, dino_cache_values = build_dino_cache_model(cfg, dino_model, train_loader_cache)
File "D:\few-shot-code\2023CaFo-main\CaFo-main\utils.py", line 96, in build_dino_cache_model
cache_keys = cache_keys.permute(1, 0)
RuntimeError: permute(sparse_coo): The number of dimensions in the tensor input does not match the length of the desired ordering of dimensions i.e. input.dim() = 4 is not equal to len(dims) = 2
Has the author encountered this problem and how to solve it?