fikadata

Did you try to increase num of epochs? I tried and it worked! Also, I have one additional question.. what is the number of epochs used in run_step1.py and run_step2.py?...