TypeError: __init__() got an unexpected keyword argument 'custom_head'

[hiromis]

This line results in "TypeError: init() got an unexpected keyword argument 'custom_head'":

learn = ts_learner(dls, InceptionTime, metrics=[mae, rmse], cbs=ShowGraph())

The cause seems to be InceptionTime's init takes **kwargs but InceptionModule does not. If you look at the signature the wildcard is in between InceptionTime's constructor arguments and InceptionModule's:

I would assume adding kwargs to InceptionModule's constructor, and possibly remove kwargs from InceptionBlock and InceptionTime's constructor will fix the issue but I am still not familiar with the code enough to say that definitely.

[oguiza]

Hi @hiromis, Can you try using InceptionTimePlus instead of InceptionTime? All models ending in Plus within tsai allow you to pass targets with different shapes. They use a custom head to recreate an output of the same shape as the target.

[hiromis]

Worked for me. Thanks!