use resnet50 to train a cifar10 classifier

2023-12-30 00:03:36 +08:00
parent 7a891969e0
commit a27d0a24d9
27 changed files with 393 additions and 19 deletions
--- a/dvclive/train/params.yaml
+++ b/dvclive/train/params.yaml
@ -0,0 +1,5 @@
+data_dir: data/processed
+epochs: 5
+batch_size: 256
+learning_rate: 5e-05
+epoch: 4