search

PaddlePaddle / PaddleVideo

Awesome video understanding toolkits based on PaddlePaddle. It supports video data annotation tools, lightweight RGB and skeleton based action recognition model, practical applications for video tagging and sport action detection.
Apache License 2.0
1.49k stars 374 forks source link

在进行篮球模型预测的时候用提供的数据和模型显示内存分配不足，怎么调整？ #577

Open czzzy opened 1 year ago

czzzy commented 1 year ago

File "action_detect\reader\audio_reader.py", line 61, in create_reader examples = feature_extractor.wav_to_example(audio_data, self.sample_rate) File "action_detect\mfcc\feature_extractor.py", line 132, in wav_to_example upper_edge_hertz=vgg_params.MEL_MAX_HZ) File "action_detect\mfcc\feature_extractor.py", line 100, in log_mel_spectrogram window_length=window_length_samples) File "action_detect\mfcc\feature_extractor.py", line 36, in stft_magnitude windowed_frames = frames * window numpy.core._exceptions.MemoryError: Unable to allocate 1000. MiB for an array with shape (204900, 640) and data type float64

huangjun12 commented 1 year ago

减少推理的batch_size，或者使用AIStudio提供的算力