File tree 1 file changed +1
-15
lines changed
1 file changed +1
-15
lines changed Original file line number Diff line number Diff line change @@ -90,6 +90,7 @@ def test_xbm_multisupcon_val(self):
90
90
self .test_xbm_multisupcon_val_gt [dtype ][b ],
91
91
atol = 1e-2 if dtype == torch .float16 else 1e-4 ))
92
92
93
+
93
94
def test_with_no_valid_pairs (self ):
94
95
for dtype in TEST_DTYPES :
95
96
embedding_angles = [0 ]
@@ -105,21 +106,6 @@ def test_with_no_valid_pairs(self):
105
106
loss .backward ()
106
107
self .assertEqual (loss , 0 )
107
108
108
- def test_ (self ):
109
- for dtype in TEST_DTYPES :
110
- embedding_angles = [0 ]
111
- embeddings = torch .tensor (
112
- [angle_to_coord (a ) for a in embedding_angles ],
113
- requires_grad = True ,
114
- dtype = dtype ,
115
- ).to (
116
- TEST_DEVICE
117
- ) # 2D embeddings
118
- labels = torch .LongTensor ([[0 ]])
119
- loss = self .loss_func (embeddings , labels )
120
- loss .backward ()
121
- self .assertEqual (loss , 0 )
122
-
123
109
124
110
def test_backward (self ):
125
111
for dtype in TEST_DTYPES :
You can’t perform that action at this time.
0 commit comments