Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit ee6fdda

Browse files
author
Taylor Robie
authored
add shell script with default MLPerf parameters (tensorflow#4751)
1 parent de715b3 commit ee6fdda

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

official/recommendation/run.sh

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
#!/bin/bash
2+
3+
# Run script for MLPerf
4+
python3 ncf_main.py -dataset ml-20m -hooks "" -hr_threshold 0.9562 -train_epochs 100 -learning_rate 0.0005 -batch_size 2048 -layers 256,256,128,64 -num_factors 64
5+

0 commit comments

Comments
 (0)