search

google-research / mint

Multi-modal Content Creation Model Training Infrastructure including the FACT model (AI Choreographer) implementation.
Apache License 2.0
497 stars 85 forks source link

Something Wrong with fast_processing in input_util.py? #18

Closed hzxie closed 2 years ago

hzxie commented 2 years ago

https://github.com/google-research/mint/blob/b8f8bdfbbe3fbfa67831a2ef9bcf71a6a9e74552/mint/utils/inputs_util.py#L101-L103

It should be:

example["audio_input"] = example["audio_sequence"][start * audio_sample_rate:start * audio_sample_rate + 
                                                   audio_input_length, :]