8 lines
161 B
Python
8 lines
161 B
Python
from train.pretrain import label
|
|
|
|
|
|
def test_label():
|
|
label_test = label(20, 0.01)
|
|
|
|
assert label_test != ''
|
|
assert label_test == 'Epoch=20 Loss=0.01'
|