Skip to content

GitLab

"tests/test_apis/test_det_inferencer.py" did not exist on "fdfe3c4f8ba935ae428a8a496ce57755d5b2ea98"
Switch branch/tag
Find file Normal viewHistoryPermalink
common.py 220 Bytes
Newer Older
huchen's avatar
Init colossal, worldsize=4, 3d test需要8dcu  
huchen committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14 
import torch

TESSERACT_DIM = 2
TESSERACT_DEP = 2
BATCH_SIZE = 8
SEQ_LENGTH = 8
HIDDEN_SIZE = 8
NUM_CLASSES = 8
VOCAB_SIZE = 16
IMG_SIZE = 16


def check_equal(A, B):
    assert torch.allclose(A, B, rtol=1e-5, atol=1e-2)