Unverified Commit d211a84a authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix `EncodecModelTest::test_multi_gpu_data_parallel_forward` (#24663)



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 469f4d0c
......@@ -73,7 +73,8 @@ class EncodecModelTester:
def __init__(
self,
parent,
batch_size=13,
# `batch_size` needs to be an even number if the model has some outputs with batch dim != 0.
batch_size=12,
num_channels=2,
is_training=False,
num_hidden_layers=4,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment