Fix EncodecModelTest::test_multi_gpu_data_parallel_forward (#24663)
fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
@@ -73,7 +73,8 @@ class EncodecModelTester:
|
|||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
parent,
|
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,
|
num_channels=2,
|
||||||
is_training=False,
|
is_training=False,
|
||||||
num_hidden_layers=4,
|
num_hidden_layers=4,
|
||||||
|
|||||||
Reference in New Issue
Block a user