Marc-Antoine BERTIN

Results 4 comments of Marc-Antoine BERTIN

You have to compute it using cuda, so replace `device=torch.device('cpu')` by `device=torch.device('cuda')` or remove it as the device argument is set to "cuda" by default.

Hi, I'm trying to implement yolov5 with MLX. So, I already made some blocks as : ConvBlock, C3f, Bottleneck and SPPF (their names in ultralytics repo). But yes, MaxPooling is...

Great work :) I have many questions about it and will try to experiment it also ! 1. Why did you use a linear layer instead of 1 more KAN...

Same error than @dedev5, the solution works with python code and not with the CLI