simonfranken/mmp_wise2526_franksim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

adapts document

Browse Source
This commit is contained in:
franksim
2025-11-04 09:35:16 +01:00
parent 78bdb7155b
commit ef540f128e
3 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mmp/a3/main.py
View File

@@ -15,11 +15,11 @@ def main():
train_epochs = 10
model = MmpNet(num_classes=2).to(device=device)
dataloader_train = get_dataloader(
path_to_data="/home/ubuntu/mmp_wise2526_franksim/.data/mmp-public-3.2",
path_to_data=".data/mmp-public-3.2",
image_size=244, batch_size=32, num_workers=6, is_train=True
)
dataloader_eval = get_dataloader(
path_to_data="/home/ubuntu/mmp_wise2526_franksim/.data/mmp-public-3.2",
path_to_data=".data/mmp-public-3.2",
image_size=244, batch_size=32, num_workers=6, is_train=False
)
criterion, optimizer = get_criterion_optimizer(model=model)