Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

UNet in the register #453

Closed
wants to merge 6 commits into from
+16 −8
Closed

UNet in the register #453

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
0cff8af
UNet in the register
Joao-L-S-Almeida Feb 21, 2025
83d04e7
adjust
Joao-L-S-Almeida Feb 21, 2025
c2837df
renaming base_channels with out_channels
Joao-L-S-Almeida Feb 21, 2025
6c5510c
ASPP missing in register
Joao-L-S-Almeida Feb 21, 2025
12bf967
out_channels need a value
Joao-L-S-Almeida Feb 21, 2025
ca2dd74
register more ASPP heads
Joao-L-S-Almeida Feb 21, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
adjust 
Signed-off-by: João Lucas de Sousa Almeida <joao.l.sa.9.3@gmail.com>
@Joao-L-S-Almeida
Joao-L-S-Almeida committed Feb 21, 2025
commit 83d04e76618161304ce58c3a4e4b82b2543818d9
1 change: 0 additions & 1 deletion terratorch/models/backbones/unet.py
View file
Original file line number Diff line number Diff line change
@@ -98,7 +98,6 @@ def __init__(self,
super(UNet, self).__init__()

self.pretrained = pretrained
print(pretrained)
assert not (init_cfg and pretrained), \
'init_cfg and pretrained cannot be setting at the same time'
if isinstance(pretrained, str):