facebookresearch / fvcore

Collection of common code that's shared among different research projects in FAIR computer vision team.
Apache License 2.0
2k stars 226 forks source link

Unsupport the high dimension inputs that larger than 5. #135

Open hailong23-jin opened 1 year ago

hailong23-jin commented 1 year ago

It raises an assert error in the following code when our input shape is [B, C, H, W, h, w], whose dimension is larger than 5. Can I directly modify 5 to 6? Does it effect the final results? https://github.com/facebookresearch/fvcore/blob/b905b9bfcaba54e04fd63054315f8120d0295975/fvcore/nn/jit_handles.py#L236-L246