hanxunh / cognitivedistillation Goto Github PK

Thank you for your work and code!
When i try to train a model , i use the config you stored in ./configs. The config i set is like below:
python train.py --exp_path output --exp_config configs/celeba --exp_name celeba_rn18
But i meet some problems like:

File "train.py", line 49, in epoch_exp_stats
loss = F.cross_entropy(logits, labels, reduction='none')
File "/home/miniconda3/envs/cog/lib/python3.8/site-packages/torch/nn/functional.py", line 3026, in cross_entropy
return torch._C._nn.cross_entropy_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index, label_smoothing)
TypeError: cross_entropy_loss(): argument 'input' (position 1) must be Tensor, not list

I didn't change the code, so i wonder if i set the config in a wrong way.
If you know why this happened , please offer me some help, Thanks a lot!

Thanks for sharing your wonderful work. When I tried to run the code, it threw errors due to dependency conflicts of packages' versions: omegaconf, fairseq, hydra-core, and mlconfig. Could you please provide the requirements.txt for setting up the environment? Thanks.