Hi, nice work for this repository! Pardon, what's the requirement for hardware? Is GPU needed for regression task? Best!