Training using SUNRGBD dataset

[sreejithpai]

Hi, Thank you very much for the training script. I am currenlty trying to train using SUNRGBD dataset for both depth prediction and semantic segmentation. I have added multiple classes to config file ( chairs, floor, tables ). While training, program stopped execution due to nan values after 3 epochs. I am using 'vit_base', batch size 8, and default values for all other parameters. I have following doubts.

1. Should I update/modify any code sections to train for multiple classes segmentation?
2. I have passes ground truth depth values as such. Should I convert it to meters ?

Thanks in advance...

[realTravisYou]

Hi, Thank you very much for the training script. I am currenlty trying to train using SUNRGBD dataset for both depth prediction and semantic segmentation. I have added multiple classes to config file ( chairs, floor, tables ). While training, program stopped execution due to nan values after 3 epochs. I am using 'vit_base', batch size 8, and default values for all other parameters. I have following doubts.

1. Should I update/modify any code sections to train for multiple classes segmentation?
2. I have passes ground truth depth values as such. Should I convert it to meters ?

Thanks in advance...

Hello, have you successfully completed your work? Can you tell me how to do it, thank you very much.