From 0f403bc60db5bee67152397d826af3eca5e0d159 Mon Sep 17 00:00:00 2001 From: MSXYZ-GenAI Date: Tue, 21 Apr 2026 14:32:21 +0300 Subject: [PATCH] Add train script for DLAANet (train_dlaa.py) --- train_dlaa.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/train_dlaa.py b/train_dlaa.py index eed9daf..f2a9441 100644 --- a/train_dlaa.py +++ b/train_dlaa.py @@ -22,7 +22,7 @@ import os import glob -# Do not change this architecture (must be the same as in VideoTAADLAA.py) +# Do not change this architecture class _DLAANet(nn.Module): def __init__(self): super().__init__() @@ -76,7 +76,8 @@ class DLAADataset(Dataset): self.input_transform = transforms.Compose([ transforms.Resize((128, 128), interpolation=transforms.InterpolationMode.BICUBIC), - # Tırtık oluşturma aşaması + + # Downsampling transforms.Resize((512, 512), interpolation=transforms.InterpolationMode.NEAREST), transforms.ToTensor() ])