How can Train KAN on batches of data

[lukmanulhakeem97]

While training KAN using entire samples say 10000, end up in memory overflow issue. Instead of giving entire data at once on "dataset" dict, can we train KAN on small batches of data like normally do on ANN?

[KindXiaoming]

Hi, you can pass the batch argument in train: https://github.com/KindXiaoming/pykan/blob/master/kan/KAN.py#L761

[lukmanulhakeem97]

thanks @KindXiaoming, got it 👍

[HajiDr]

@KindXiaoming can you explain how does the batch argument works in case of KANs. It seems only training and testing on the batch size specified per step. Or i got it wrong?

[sparcycram]

@KindXiaoming

Any thoughts appreciated

I am still having problems OutOfMemoryError with large datasets. Is it possible to modify the code to use Dataloader and pass train_loader and test_loader to train instead of dataset?

So edit this line

def train(self, dataset, opt="LBFGS", steps=100, ......