Unable to run experiment with cross-entropy

thulas / dac-label-noise

Label de-noising for deep learning

58 stars 9 forks source link

Unable to run experiment with cross-entropy #4

Open RIMcKinley opened 4 years ago

RIMcKinley commented 4 years ago

Running

python train_dac.py --datadir data --dataset stl10-c --train_y train_y_downshifted_random_monkeys.bin --nesterov --net_type vggnet -use-gpu --epochs 200 --learn_epochs 20 --seed 0 fails after the first epoch, triggering a CUDA error ('device-side assert triggered')

thulas commented 4 years ago

Hard to say what's triggering this assert. Can you run this inside PDB and print out the back-trace?