Hello, Thanks for your brilliant work! When I run the code, I find the step loss always equals nan when I use the bdd dataset. After carefully checking the code, I find the last block of the upsample_block' s output will be nan. I just use the fp16 model and follow the pipeline. Could anyone tell me what is the reason? Thanks a lot!
AI Analysis
This issue appears to be discussing a feature request or bug report related to the repository. Based on the content, it seems to be resolved. The issue was opened by maobenz and has received 7 comments.
Add a comment
Comment form would go here