Any mesure to the functions memory consumption? After I replace all 34 RELU to Mish activation, the available batch size shrinks a lot.