this repo has no description
Remove debug prints from modeling_deepseekocr2.py
BASE/PATCHES shape prints fired on every forward pass during training.
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
+2
-8
+2
-8
src/deepseek_ocr2/modeling_deepseekocr2.py
+2
-8
src/deepseek_ocr2/modeling_deepseekocr2.py
···
439
439
global_features = global_features_2
440
440
global_features = self.projector(global_features)
441
441
442
-
print('=====================')
443
-
print('BASE: ', global_features.shape)
444
-
print('PATCHES: ', local_features.shape)
445
-
print('=====================')
442
+
pass
446
443
447
444
_, hw, n_dim = global_features.shape
448
445
# h = w = int(hw ** 0.5)
···
471
468
global_features_2 = qwen2_model(global_features_1)
472
469
global_features = global_features_2
473
470
global_features = self.projector(global_features)
474
-
print('=====================')
475
-
print('BASE: ', global_features.shape)
476
-
print('NO PATCHES')
477
-
print('=====================')
471
+
pass
478
472
_, hw, n_dim = global_features.shape
479
473
# h = w = int(hw ** 0.5)
480
474