Comments (3)
Do you follow the version of python and pytorch? You can do a transformation (Long->Float) according to this error here. Meanwhile, the CE loss is just shown here and not included during the model training. You can directly delete these related codes.
from mc-net.
After I commit the loss_seg, there is a new error message:
Traceback (most recent call last): File "/home/zhaoyan/MC-Net/./code/train_mcnet_3d.py", line 147, in <module> loss_seg_dice += dice_loss(y_prob[:,1,...], label_batch[:labeled_bs,...] == 1) File "/home/zhaoyan/MC-Net/code/utils/losses.py", line 5, in Binary_dice_loss intersection = 2 * torch.sum(predictive * target) + ep RuntimeError: The size of tensor a (2) must match the size of tensor b (112) at non-singleton dimension 1
from mc-net.
The tensot a is `tensor([[[0.0094, 0.0129, 0.0148, ..., 0.0104, 0.0110, 0.0102],
[0.0078, 0.0092, 0.0045, ..., 0.0089, 0.0092, 0.0069],
[0.0054, 0.0058, 0.0087, ..., 0.0116, 0.0114, 0.0071],
...,
[0.0087, 0.0107, 0.0076, ..., 0.0150, 0.0129, 0.0111],
[0.0094, 0.0073, 0.0089, ..., 0.0057, 0.0117, 0.0067],
[0.0085, 0.0109, 0.0082, ..., 0.0091, 0.0103, 0.0077]],
[[0.0026, 0.0038, 0.0028, ..., 0.0045, 0.0039, 0.0065],
[0.0071, 0.0027, 0.0018, ..., 0.0040, 0.0035, 0.0042],
[0.0011, 0.0021, 0.0004, ..., 0.0037, 0.0030, 0.0043],
...,
[0.0095, 0.0077, 0.0096, ..., 0.0067, 0.0060, 0.0072],
[0.0070, 0.0137, 0.0070, ..., 0.0090, 0.0042, 0.0060],
[0.0094, 0.0063, 0.0086, ..., 0.0103, 0.0050, 0.0078]]],
device='cuda:0', grad_fn=<SelectBackward0>)`
The tensor b is:
`tensor([[[[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
...,
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False]],
[[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
...,
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False]],
[[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
...,
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False]],
...,
[[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
...,
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False]],
[[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
...,
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False]],
[[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
...,
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False]]],
[[[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
...,
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False]],
[[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
...,
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False]],
[[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
...,
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False]],
...,
[[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
...,
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False]],
[[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
...,
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False]],
[[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
...,
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False]]]], device='cuda:0')`
from mc-net.
Related Issues (16)
- Did you crop raw samples on the z axis? HOT 1
- 请问可以提供预处理好的Pancreas数据吗? HOT 1
- Asking for the CT-82 dataset HOT 1
- Pancreas dataset preprocessing success,but the DICE score fluctuated around 0.2 to 0.1 during training. HOT 4
- 关于Multi-scale MC-Net+和sharpening函数问题请教 HOT 1
- About the parameter quantity in the MC-Net+ HOT 2
- Number of classes issue HOT 1
- Labels for unlabeled data HOT 3
- 训练标签问题 HOT 1
- train_mcnet_2d.py HOT 3
- Loss function mistake HOT 1
- About the consistency_weight HOT 1
- Data preparation for ACDC, LA and Pancreas HOT 6
- Unmatched tensors HOT 12
- Train VNet with all 80 labels HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
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.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from mc-net.