Cannot cast array data from dtype('O') to dtype('float64') according to the rule 'safe'

kiddyboots216 / CommEfficient

PyTorch for benchmarking communication-efficient distributed SGD optimization algorithms

72 stars 20 forks source link

Open Antonio-demo opened 1 year ago

Antonio-demo commented 1 year ago

It happens this error when I run cv_train.py.

kiddyboots216 commented 1 year ago

I'm unable to reproduce this error. Have you changed the definition of lambda_step?

Antonio-demo commented 1 year ago

Can you share the source code with me? At least, I want to know where the problem happens.

kiddyboots216 commented 1 year ago

The source code is this GitHub repository.