Skip to content

Commit e6a2eb2

Browse files
committed
BUG: remove printing.
1 parent 1ae744c commit e6a2eb2

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

barcodebert/pretraining.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -804,7 +804,6 @@ def train_one_epoch(
804804
ct_forward = torch.cuda.Event(enable_timing=True)
805805
ct_forward.record()
806806
# Perform the forward pass through the model
807-
print(config.arch)
808807
if config.arch == "maelm":
809808
# print("MAELM is implemented")
810809
out = model(masked_input, att_mask, masked_unseen_tokens, config.maelm_version)

0 commit comments

Comments
 (0)