KindXiaoming / pykan

Kolmogorov Arnold Networks
MIT License
15.1k stars 1.4k forks source link

auto_save=self.auto_save in model.prune() #475

Open yixi0527 opened 1 month ago

yixi0527 commented 1 month ago

change model2=MultKAN(..., auto_save=Ture, ...) to auto_save=self.auto_save, as when we set auto_save=False, then prune the model, it'll be saved and file './model/history.txt' can't be found

362

dps123 commented 1 month ago

_pruneinput method has the same problem.