tdpeuter/2025ML-project-neural_compression
tdpeuter/2025ML-project-neural_compression
Archived
1
Fork 0
Code Issues Pull requests Releases 1 Activity Actions

feat: updates to datasets/-loaders

Browse source
This commit is contained in:
RobinMeersman 2025-11-27 19:26:59 +01:00
parent ed44d5b283
commit d2e6d17f55
11 changed files with 105 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

6
CNN-model/trainers/FullTrainer.py
View file

@ -4,9 +4,9 @@ import torch
from torch import nn as nn
from torch.utils.data import DataLoader
from trainer import Trainer
from train import train
from ..utils import print_losses
from .trainer import Trainer
from .train import train
from utils import print_losses
class FullTrainer(Trainer):
def execute(

6
CNN-model/trainers/OptunaTrainer.py
View file

@ -6,9 +6,9 @@ import torch
from torch import nn as nn
from torch.utils.data import DataLoader
from trainer import Trainer
from ..model.cnn import CNNPredictor
from train import train
from .trainer import Trainer
from model.cnn import CNNPredictor
from .train import train
def create_model(trial: tr.Trial, vocab_size: int = 256):

5
CNN-model/trainers/__init__.py
View file

@ -1,2 +1,3 @@
from OptunaTrainer import OptunaTrainer
from trainer import Trainer
from .OptunaTrainer import OptunaTrainer
from .FullTrainer import FullTrainer
from .trainer import Trainer