manuelcugliari comments

Repositories
Issues
Comments

Results 2 comments of


                                            manuelcugliari

running example 10 with GPU, error: all tensors on two devices, cuda:0 and cpu!

Same issue also specifying the device parameter. device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') model = KAN(width=[X_train_torch.shape[1], 10, 2], grid=3, k=3, device=device)

symbolic_formula is broken

Same issue (maybe). When I try to get the the symbolic formula (model.symbolic_formula(var=input_features) I get 'ZeroDivisionError: integer division or modulo by zero'