Jordan Totten

## Expected Behavior In [mlops_data_management.ipynb](https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/ml_ops/stage1/mlops_data_management.ipynb), expecting the dataflow pipeline to run successfully ## Actual Behavior get error: ``` KeyError: "tip_bin [while running 'Write TF Test Data/Write/WriteImpl/WriteBundles-ptransform-559']" ``` ## Steps to...

Guidance on evaluating shard count/size, WebDataset length

2

Trying to optimize TPU pod slice solution discussed in https://github.com/webdataset/webdataset/issues/47 Assuming scenario where you need to replicate training with epochs i.e., declare length of dataset, distribute equal samples/batches to nodes......

Resized Dataset for multiprocessing data loading

9

I'm trying to use webdataset for a distributed Pytorch XLA POC. I tried implementing the `ResizedDataset` class but start receiving many errors like the following after ~40 training steps. Any...

[BUG] merlin models on vertex ai training - cuda error

6

**Describe the bug** I've been able to train a Merlin model in a Vertex Notebook (using merlin base image). Now I'm trying to train the same model in Vertex AI...

Errors when saving ranking policies

2

What is recommended way to save ranking policies? For each agent policy in the [ranking tutorial](https://www.tensorflow.org/agents/tutorials/ranking_tutorial#vector_feedback), im getting errors like below when trying to save with [PolicySaver](https://github.com/tensorflow/agents/blob/498923f9dfbb5193249c1fe1cadb1cdd03f391a9/tf_agents/policies/policy_saver.py#L103) ``` saver =...