milesial comments

Repositories
Issues
Comments

Results 62 comments of


                                            milesial

The output image is strange : train.py or predict.py problem?

Can you share the pretrained model along with the input image that results in these masks?

Training on mac mps instead of cuda

Hi, can you try the latest master? I've added a check for MPS device in the autocast. But since autocast only supports CPU and CUDA, you should still turn AMP...