Machine-Learning
/
aiuNN
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

increased image size

This commit is contained in:
Falko Victor Habel 2025-02-15 13:29:50 +01:00
parent 4c84086932
commit 62825e9731
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/aiunn/finetune.py
View File

@ -18,7 +18,7 @@ from torch.utils.checkpoint import checkpoint
class aiuNNDataset(torch.utils.data.Dataset):
def __init__(self, parquet_path):
self.df = pd.read_parquet(parquet_path, columns=['image_512', 'image_1024']).head(2000)
self.df = pd.read_parquet(parquet_path, columns=['image_512', 'image_1024']).head(10000)
self.augmentation = Compose([
RandomBrightnessContrast(p=0.5),
HorizontalFlip(p=0.5),