yeah i have done that and there wasn’t anything problematic though i think i should use a relu between adaptiveavgpool2d and softmax
