Commit a191b9c4 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI

Fix error on numpy zero creation

parent bf9ccd78
Pipeline #29926 passed with stage
in 12 minutes and 1 second
......@@ -35,8 +35,8 @@ def _set_average(trainer, trainers, data):
if isinstance(trainer, KMeansTrainer):
# K-Means statistics
trainer.zeroeth_order_statistics = numpy.zeros(0., trainer.zeroeth_order_statistics.shape)
trainer.first_order_statistics = numpy.zeros(0., trainer.first_order_statistics.shape)
trainer.zeroeth_order_statistics = numpy.zeros(trainer.zeroeth_order_statistics.shape)
trainer.first_order_statistics = numpy.zeros(trainer.first_order_statistics.shape)
trainer.average_min_distance = 0.
for t in trainer:
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment