Problems with reproducing the LAME results

[kdiAAA]

Hello! I tried to reproduce the results of your paper in Fig. 4 for the baseline method and the LAME method, but I always get slightly different results compared to the output on paper. I have tried different batch sizes (16.64) but most of the problems are in the I.I.D. with Likelihood Shift + Prior Shift scenario. I also shared the results of each experiment with different batch sizes. Could you suggest how can I solve this problem? Because I want to try your approach for my experiments and I need to reproduce the results like in the post.

Thanks in advance!