The paper mentioned that training can be completed with just one A6000. I would like to ask how much gpu memory is used?