search

hellowangqian / sganet-crowd-counting

Pytorch implementation of SGANet for crowd counting
20 stars 5 forks source link

Error while training #2

Open bhawanatyagi opened 3 years ago

bhawanatyagi commented 3 years ago

The error is encountering while training in the below line epoch_loss = running_loss / dataset_sizes['train'] error is :float division by zero

alercelik commented 2 years ago

Same happened to me. Make sure you entered data_dir correctly. And there is a bit of an error in code where data_dir is provided to ShanghaiTechDataset. Correct it as well.

bhawanatyagi commented 1 year ago

Thank u So Much. -- Thanks and Regards!

Bhawana Tyagi

On Fri, Mar 25, 2022 at 6:59 PM alercelik @.***> wrote:

Same happened to me. Make sure you entered data_dir correctly. And there is a bit of an error in code where data_dir is provided to ShanghaiTechDataset. Correct it as well.

— Reply to this email directly, view it on GitHub https://github.com/hellowangqian/sganet-crowd-counting/issues/2#issuecomment-1079029025, or unsubscribe https://github.com/notifications/unsubscribe-auth/ARFLLSHBT3FYZLWSBSK3GF3VBW5SJANCNFSM5BIFOI3Q . You are receiving this because you authored the thread.Message ID: @.***>