Skip to content

Commit

Permalink
fix unrelated pre commit bug
Browse files Browse the repository at this point in the history
  • Loading branch information
kpertsch committed Mar 11, 2024
1 parent a8e5875 commit db78206
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions examples/06_pytorch_oxe_dataloader.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,13 +4,14 @@
shuffle buffer size, which are reduced for demonstration purposes).
"""
import numpy as np
from octo.data.dataset import make_interleaved_dataset
from octo.data.oxe import make_oxe_dataset_kwargs_and_weights
import tensorflow as tf
import torch
from torch.utils.data import DataLoader
import tqdm

from octo.data.dataset import make_interleaved_dataset
from octo.data.oxe import make_oxe_dataset_kwargs_and_weights

DATA_PATH = "gs://rail-orca-central2/resize_256_256"

tf.config.set_visible_devices([], "GPU")
Expand Down

0 comments on commit db78206

Please sign in to comment.