neulus/cloud-removal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

pixel shuffle

Browse Source
This commit is contained in:
neulus
2025-10-01 16:30:05 +09:00
parent 8966cafb8f
commit 49025c4d87
19 changed files with 162 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
main.py
View File

@@ -11,7 +11,7 @@ from src.dataset.preprocess import denormalize
from src.model.utransformer import UTransformer
from src.rf import RF
device = "cuda:0"
device = "cuda:2"
model = UTransformer.from_pretrained_backbone(
"facebook/dinov3-vitl16-pretrain-sat493m"
@@ -21,7 +21,7 @@ optimizer = optim.AdamW(model.parameters(), lr=1e-4)
train_dataset, test_dataset = get_dataset()
wandb.init(project="cloud-removal-kmu", id="icy-field-11", resume="allow")
wandb.init(project="cloud-removal-kmu", id="icy-field-12", resume="allow")
if not (wandb.run and wandb.run.name):
raise Exception("nope")
@@ -37,7 +37,7 @@ if os.path.exists(checkpoint_path):
start_epoch = checkpoint["epoch"] + 1
batch_size = 4
accumulation_steps = 4
accumulation_steps = 8
total_epoch = 1000
for epoch in range(start_epoch, total_epoch):
lossbin = {i: 0 for i in range(10)}