Commit e5a1876
authored
1132-fix-the-bug-in-loading-tool-segmentation-model (#1135)
Signed-off-by: binliu <binliu@nvidia.com>
Fixes #1132 .
### Description
The pretrained `endoscopic_tool_segmentation` model in MONAI model zoo
cannot be loaded to `video_seg` tutorial since the different output.
Therefore, the whole model except segmentation head should be loaded to
perform pretrain.
### Checks
<!--- Put an `x` in all the boxes that apply, and remove the not
applicable items -->
- [ ] Notebook runs automatically `./runner [-p <regex_pattern>]`
Signed-off-by: binliu <binliu@nvidia.com>1 parent cbeb69c commit e5a1876
1 file changed
+6
-7
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
372 | 372 | | |
373 | 373 | | |
374 | 374 | | |
375 | | - | |
| 375 | + | |
376 | 376 | | |
377 | | - | |
378 | | - | |
379 | | - | |
380 | | - | |
| 377 | + | |
| 378 | + | |
| 379 | + | |
381 | 380 | | |
382 | 381 | | |
383 | 382 | | |
| |||
691 | 690 | | |
692 | 691 | | |
693 | 692 | | |
694 | | - | |
| 693 | + | |
695 | 694 | | |
696 | | - | |
| 695 | + | |
697 | 696 | | |
698 | 697 | | |
699 | 698 | | |
| |||
0 commit comments