We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6595d7b commit 2944a9dCopy full SHA for 2944a9d
run_python_examples.sh
@@ -155,10 +155,11 @@ function vision_transformer() {
155
156
function word_language_model() {
157
uv run main.py --epochs 1 --dry-run $CUDA_FLAG --mps || error "word_language_model failed"
158
+ uv run generate.py $CUDA_FLAG --mps || error "word_language_model generate failed"
159
for model in "RNN_TANH" "RNN_RELU" "LSTM" "GRU" "Transformer"; do
160
uv run main.py --model $model --epochs 1 --dry-run $CUDA_FLAG --mps || error "word_language_model failed"
161
162
done
- uv run generate.py $CUDA_FLAG --mps || error "word_language_model generate failed"
163
}
164
165
function gcn() {
0 commit comments