Aminul Hoque

Results 3 comments of Aminul Hoque

> A workaround seems to do the trick for me. Add those lines before `apt-get update` > > ``` > RUN rm /etc/apt/sources.list.d/cuda.list > RUN rm /etc/apt/sources.list.d/nvidia-ml.list > ``` >...

Facing the same problem. Any solution/thoughts on this?

This example runs for PyTorch, not for Tensorflow. To work with Tensorflow, we need to "batchify" the data. `test_dataset = dataset.get_split('validation') test_split = TFDataloader(dataset=test_dataset, model=model, use_cache=False) test_loader, len_test = test_split.get_loader(batch_size=1,transform=False)...