search

Lordsedion / Transformer-Refactor

0 stars 0 forks source link

Config file #2

Open Lordsedion opened 2 months ago

Lordsedion commented 2 months ago

Train and infrence

Lordsedion commented 2 months ago

Train: Input = train data, where to save model. Output = Validation and test set performance , model output Inference = some dataset (or data loader) <- make it able to take both, where to save the predictions, Output = predictions Inference_p just take from line 306 and below for now