search

KaiyangZhou / ssdg-benchmark

Benchmarks for semi-supervised domain generalization.
MIT License
67 stars 9 forks source link

Key error:"img0" #1

Closed knight-fzq closed 3 years ago

knight-fzq commented 3 years ago

Dear author, When I follow your directions and run the command:bash run_ssdg.sh ssdg_pacs 210 v1, there is a error that Key error:"Img0". So I ran the command print(batch_x) and I found that there was no key called "img0". Only "img" and "img2" in that dict. Here is the shot screen: Screenshot from 2021-08-06 13-51-38

KaiyangZhou commented 3 years ago

Oops, I upaded the underlying source code in Dassl.pytorch by adding DATALOADER.RETURN_IMG0, which decides whether to return un-augmented images, but forgot to update the config files here. See here for more details.

I have updated the config files in this repo so the error is fixed. Please pull the latest commit.

knight-fzq commented 3 years ago

Thank you!!!