Magical Potato

@Ruyii2 No not yet, ill let you know if i figure anything out :/

@lnhutnam Once i get back to working with yolo i will try changing the torch version, i think you are right and its probably just a version issue. :)