test transforms are used instead of predict transforms

In the COCO datamodule, predict_transform_begin and predict_transform_end are not used and their "test" versions are used instead.