Deep learning

Current implementation of Go binding can not specify options.

GPUOptions struct is in internal package. And go generate doesn't work for protobuf directory. So we can't specify GPUOptions for NewSession.

Model description

I would like to add a new model:

Proposed in the paper: UNETR: Transformers for 3D Medical Image Segmentation

UNEt TRansformers (UNETR) utilize a transformer as the encoder to learn sequence representations of the input volume and effectively capture the global multi-scale information, while also following the successful "U-shaped"

Followup to pytorch/pytorch#74955 (comment).

It turns out that that cmake version was just bad and we can now unpin cmake once again.

cc @seemethere @malfet @pytorch/pytorch-dev-infra

What happened + What you expected to happen

The train_loop_per_worker code snippets contain train.get_world_size(), train.get_world_rank(), and train.get_local_rank(). But, those functions don't exist. The actual names are train.world_size(), train.world_rank() and train.local_rank().

Versions / Dependencies

N/A

Reproduction script

See:

• https://docs.ray.io/en/mast