How to train on custom dataset

Hi, thanks for the great work.

May I ask how can I train the model on my own dataset? From the paper, seems there is one step to extract coarse light and albedo information, just want to know how I can do that for my own dataset.

Thanks in advance.