error 'axis'

[akshayvijayan]

On running pretrained model the following error occurs
TypeError: softmax() got an unexpected keyword argument 'axis'
Please help to correct the same.

[someshium]

@akshayvijayan Did you Got the solution to this or can you please provide some other repo from where i can take the reference?
thanks