RGBTask: First code with working Torch mode