Marc-Antoine BERTIN
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