digital-forensics-labs/Machine-Learning-Collection
1
0
Fork 0
mirror of https://github.com/aladdinpersson/Machine-Learning-Collection.git synced 2026-04-10 12:33:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

add lightning code, finetuning whisper, recommender system neural collaborative filtering

Browse Source
This commit is contained in:
Aladdin Persson
2023-02-21 16:25:42 +01:00
parent c646ef65e2
commit 94f6c024fe
51 changed files with 17977 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
ML/Pytorch/pytorch_lightning/7. Callbacks/config.py Normal file
View File

@@ -0,0 +1,15 @@
# Training hyperparameters
INPUT_SIZE = 784
NUM_CLASSES = 10
LEARNING_RATE = 0.001
BATCH_SIZE = 64
NUM_EPOCHS = 1000
# Dataset
DATA_DIR = "dataset/"
NUM_WORKERS = 4
# Compute related
ACCELERATOR = "gpu"
DEVICES = [0]
PRECISION = 16