Can we provide custom epoch value in training_1_epoch.sh file by replacing any value in-place of 1

intel / ai-reference-models

Intel® AI Reference Models: contains Intel optimizations for running deep learning workloads on Intel® Xeon® Scalable processors and Intel® Data Center GPUs

Apache License 2.0

683 stars 220 forks source link

Can we provide custom epoch value in training_1_epoch.sh file by replacing any value in-place of 1 #163

Open Dirojit opened 11 months ago

Dirojit commented 11 months ago

--steps=100 train_epochs=1 epochs_between_evals=1

Can we change to "--steps=100 train_epochs=10 epochs_between_evals=1" in above line

sramakintel commented 7 months ago

@Dirojit which workload are you trying to run custom epochs for?