I am trying to customize YOLO architecture to accept 4 channel RGBD input. I added ch:4 to the .yaml architecture file, and then created a custom dataloader to handle the incoming images (saved as ...

I am new to accelerate, and I am testing the checkpointing stuff. I have this test script which basically testing if dataloader wraped by acclerator will produce the same shuffled dataset order ...