Error in "python train.py"

allenai / unifew

Unifew: Unified Fewshot Learning Model

Apache License 2.0

18 stars 3 forks source link

Open code-wanderlust opened 2 years ago

code-wanderlust commented 2 years ago

Hello, I just run the below command

"CUDA_VISIBLE_DEVICES=0 python train.py hydra.run.dir=tmp/model/ trainer.max_steps=30000 query_batch_size=4 +sampler.min_way=2"

But got a error message, "Peeking at some test datasets!!!!" in "unifew/utils.py", "assemble_sampler_cfg".

How can I fix this issue ?

armancohan commented 2 years ago

I'm not able to reproduce this. Are you are trying to run the exact code as it appears on this repo or you've made some changes?