infer

[1390806607]

Hello, may I ask that the shape of your inference code output is [b,2,w,h]. Why only use the -1 dimension

[1390806607]

        if args.version == 'new':
            enslist.append(sample[:,-1,:,:])
        else:
            enslist.append(co)

[Issues-translate-bot]

Bot detected the issue body's language is not English, translate it automatically. 👯👭🏻🧑‍🤝‍🧑👫🧑🏿‍🤝‍🧑🏻👩🏾‍🤝‍👨🏿👬🏿

---

if args.version == 'new': enslist.append(sample[:,-1,:,:]) else: enslist.append(co)