Commit 76142ec
authored
Update group & priority tags in unet model (#2961)
### Ticket
- Fixes #2960
### Problem description
- UNet tests in tt-forge-fe using two UNet implementations currently.
The goal is to determine which one mimics the [target tensorflow
implementation](https://github.com/hassanlougithub/brain-tumor-seg) and
keep only that test in red.
-
[vgg19_unet/pytorch/loader.py](https://github.com/tenstorrent/tt-forge-models/blob/main/vgg19_unet/pytorch/loader.py)
-
[unet/pytorch/loader.py](https://github.com/tenstorrent/tt-forge-models/blob/57faf96e6050915e4fc6f22454e32609479ca4b6/unet/pytorch/loader.py#L69C13-L69C63)
### What's changed
- Red tag removed from the test using
[unet/pytorch/loader.py](https://github.com/tenstorrent/tt-forge-models/blob/57faf96e6050915e4fc6f22454e32609479ca4b6/unet/pytorch/loader.py#L69C13-L69C63)
as it fails to mimic target implementation
- Added red tag to the test using
[vgg19_unet/pytorch/loader.py](https://github.com/tenstorrent/tt-forge-models/blob/main/vgg19_unet/pytorch/loader.py)
as it exactly matches the target TensorFlow implementation. for more
context refer -
tenstorrent/tt-forge-models#152 (comment)
### Checklist
- [x] New/Existing tests provide coverage for changes1 parent 6bf215c commit 76142ec
File tree
2 files changed
+2
-2
lines changed- forge/test/models/pytorch/vision/unet
2 files changed
+2
-2
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
88 | 88 | | |
89 | 89 | | |
90 | 90 | | |
91 | | - | |
92 | | - | |
93 | 91 | | |
94 | 92 | | |
95 | 93 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
32 | 32 | | |
33 | 33 | | |
34 | 34 | | |
| 35 | + | |
| 36 | + | |
35 | 37 | | |
36 | 38 | | |
37 | 39 | | |
| |||
0 commit comments