machine-learning-deep-learning/Machine-Learning-Collection
1
0
Fork 0
mirror of https://github.com/aladdinpersson/Machine-Learning-Collection.git synced 2026-02-21 11:18:01 +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

12
ML/Pytorch/pytorch_lightning/7. Callbacks/callbacks.py Normal file
View File

@@ -0,0 +1,12 @@
from pytorch_lightning.callbacks import EarlyStopping, Callback
class MyPrintingCallback(Callback):
def __init__(self):
super().__init__()
def on_train_start(self, trainer, pl_module):
print("Starting to train!")
def on_train_end(self, trainer, pl_module):
print("Training is done.")