Spaces:

SteveaWong
/

AI-Drawing-Predictor

Sleeping

App Files Files Community

AI-Drawing-Predictor / training

16.9 kB

Ctrl+K

Ctrl+K

2 contributors

History: 2 commits

SteveWong-a

Add dropout and weight decay to prevent overfitting

7150265 4 months ago

dataset.py

5.93 kB
Update train.py with Hugging Face Hub upload capabilities 4 months ago
test_inference.py

3.11 kB
Update train.py with Hugging Face Hub upload capabilities 4 months ago
train.py

7.86 kB
Add dropout and weight decay to prevent overfitting 4 months ago