Please make CPU support

[Land-Maze]

I tried to change file train_det.py (36 line) to

@classmethod
    def build_model(cls, cfg):
        cfg.defrost()
        cfg.MODEL.DEVICE = "cpu"
        cfg.freeze()
        model = build_model(cfg)
        return model

But in file yolov7.py there's

mask, obj_mask = mask.cuda(), obj_mask.cuda()

Which kills option to try it with CPU
I want to test my M1

[Land-Maze]

Oops, i tried to change all cuda() to cpu() and it worked
In yolov7.py and yolov7p.py