Fix EncodecModelTest::test_multi_gpu_data_parallel_forward (#24663)

fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2023-07-05 11:37:46 +02:00
committed by GitHub
parent 469f4d0c29
commit d211a84aca

View File

@@ -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,