try with mutliclass

[Adithia88]

i was tried using multiclass and got some error

find learning rate to train the last layer group first

fucntion when error :
learn.freeze()
learn.lr_find()
learn.recorder.plot()

error message : can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

i search in stackoverflow too but still got error , i read some link and its said , cause i use multiclass , and your example in binary

what should i do ?

[naity]

Hi there,

When you call CNNTabularModel, did you set out_sz as the number of classes you have? Thanks