ValueError: Input X contains NaN occured when running MNIST task with RFLBAT defense

haoyangliASTAPLE / 3DFed

36 stars 4 forks source link

ValueError: Input X contains NaN occured when running MNIST task with RFLBAT defense #5

Open nguyenhongson1902 opened 11 months ago

nguyenhongson1902 commented 11 months ago

Hi, I was running python training.py --name mnist --params configs/mnist_fed.yaml with 3DFed attack, RFLBAT defense then came up with this error (image below)

Can you help me to review the code and fix the bug?