mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-17 13:12:19 +00:00
minor fix
This commit is contained in:
parent
cf796eefed
commit
a6eead6c98
@ -747,7 +747,7 @@ def get_model(params: AttributeDict) -> nn.Module:
|
|||||||
|
|
||||||
|
|
||||||
def get_spec_augment(params: AttributeDict) -> SpecAugment:
|
def get_spec_augment(params: AttributeDict) -> SpecAugment:
|
||||||
num_frame_masks = 10 * params.time_mask_ratio
|
num_frame_masks = int(10 * params.time_mask_ratio)
|
||||||
max_frames_mask_fraction = 0.15 * params.time_mask_ratio
|
max_frames_mask_fraction = 0.15 * params.time_mask_ratio
|
||||||
logging.info(
|
logging.info(
|
||||||
f"num_frame_masks: {num_frame_masks}, "
|
f"num_frame_masks: {num_frame_masks}, "
|
||||||
|
Loading…
x
Reference in New Issue
Block a user