Git Product home page Git Product logo

mae_segmentation's Issues

执行训练脚本时报错，不知道是否为torch版本的原因

你好，想问下项目代码运行的torch版本是多少呢？

我使用torch==1.10.0，有以下的错误，把continuous操作修改为clone以及relu修改为inplace=False，还是会报相同的问题(训练的过程中，我设置了use_fp16=False)

RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor [2, 768, 32, 32]], which is output 0 of ReluBackward0, is at version 1; expected version 0 instead. Hint: enable anomaly detection to find the operation that failed to compute its gradient, with torch.autograd.set_detect_anomaly(True).

model weight

Dear,

Thanks for your great work!

With your offered code and hyper-parameters, I get the results as follows:

2022-10-01 04:03:09,229 - mmseg - INFO - Iter(val) [16000]      mIoU: 0.3869, mAcc: 0.5037, aAcc: 0.8005
2022-10-01 05:56:34,575 - mmseg - INFO - Iter(val) [32000]      mIoU: 0.4353, mAcc: 0.5557, aAcc: 0.8148
2022-10-01 07:49:43,813 - mmseg - INFO - Iter(val) [48000]      mIoU: 0.4535, mAcc: 0.5794, aAcc: 0.8188
2022-10-01 09:42:33,149 - mmseg - INFO - Iter(val) [64000]      mIoU: 0.4523, mAcc: 0.5758, aAcc: 0.8216
2022-10-01 11:35:34,234 - mmseg - INFO - Iter(val) [80000]      mIoU: 0.4655, mAcc: 0.5783, aAcc: 0.8256
2022-10-01 13:28:33,442 - mmseg - INFO - Iter(val) [96000]      mIoU: 0.4648, mAcc: 0.5726, aAcc: 0.8279
2022-10-01 15:21:28,416 - mmseg - INFO - Iter(val) [112000]     mIoU: 0.4678, mAcc: 0.5798, aAcc: 0.8252
2022-10-01 17:14:35,033 - mmseg - INFO - Iter(val) [128000]     mIoU: 0.4683, mAcc: 0.5806, aAcc: 0.8270
2022-10-01 19:07:43,025 - mmseg - INFO - Iter(val) [144000]     mIoU: 0.4729, mAcc: 0.5804, aAcc: 0.8279
2022-10-01 21:00:48,207 - mmseg - INFO - Iter(val) [160000]     mIoU: 0.4758, mAcc: 0.5841, aAcc: 0.8293

It seems a few lower than yours.

Could you provide the model weights that reach 48.1% ?
Sincerely.

Visualization or demo?

Thanks for your effort in sharing this excellent work. Can you please provide a demo of applying the pre-trained models to custom images?
Does the network apply masking during training and testing like MAE?

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.