RICO Training

Hey, 
Thank you for releasing this code!

Any pointers on which rico dataset to use and how to preprocess it to feed into the transformer?
Are you starting with "RICO UI SCREENSHOTS AND HIERARCHIES WITH SEMANTIC ANNOTATIONS"?

Thanks!