Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Unable to benchmark pre-trained image model: freezing on import

See original GitHub issue

Hello, thanks for providing the code for your paper!

I have been unable to run either the benchmark_agent.py code (with the pre-trained model-10.th), or the data_collector.py script. Both scripts seem to be freezing at some point in the import process, which I haven’t been able to resolve. They don’t throw errors (or print anything besides the pygame message) but seem to hang indefinitely.

I had to make a couple of modifications to the installed dependencies, because it seems that the RTX GPU family is not compatible with CUDA 9.0 or below (or with any packages that are built with CUDA 9.0 or below), as per https://github.com/pytorch/pytorch/issues/17543, and because I ran into CUDA warnings and CUDNN errors when using the original dependencies (that were built with CUDA 8.0)

The relevant versions of dependencies on my system are as follows:

Package	Version
cudatoolkit	10.1.243
pytorch	py3.5_cuda10.1.243_cudnn7.6.3_0
cudnn	7.6.5

If anyone has guidance on how to debug/resolve this issue, I’d really appreciate it. Thanks so much!

My set-up: GPU: GeForce RTX 2080 with Max-Q CUDA version: 10.1

Issue Analytics

State:
Created 4 years ago
Comments:13 (2 by maintainers)

Top GitHub Comments

2reactions

rohanb2018commented, Feb 5, 2020

Great, I think I actually got the benchmark_agent.py example working now!

I ended up commenting out all of the torchvision imports inside the bird_view folder. This included replacing all of the ToTensor calls with the code that you mentioned. Additionally, I had to comment out instances of torchvision.utils inside the logger.py and saver.py files. I replaced the calls to tv_utils.make_grid in both of those files by just copying the source code from PyTorch (https://pytorch.org/docs/stable/_modules/torchvision/utils.html#make_grid).

I assume I’ll have to do the same to the files inside the training folder since I see some references to torchvision there as well.

Thanks again for your help! Will let you know if I run into any other issues.

1reaction

dotchencommented, Jul 15, 2020

@peiyunh start_dtcrowd/stop_dtcrowd only come with our custom CARLA 0.9.6 egg for the pedestrian fix. If you would like to use this repo with CARLA 0.9.9 you need to modify some of the utilities code.

Top Results From Across the Web

Transfer learning from pre-trained models | by Pedro Marcelino

This article teaches you how to use transfer learning to solve image classification problems. A practical example using Keras and its pre- ......

Fine-tune a pretrained model - Hugging Face

Don't worry, this is completely normal! The pretrained head of the BERT model is discarded, and replaced with a randomly initialized classification head....

How to use Deep Learning Models from Keras for a problem ...

It is not necessary to implement a model from scratch to freeze a few layers. You can do this on pre-trained models as...

Save, Load and Inference From TensorFlow Frozen Graph

TensorFlow model saving has become easier than it was in the early days. Now you can either use Keras to save h5 format...

Hands-on Transfer Learning with Keras and the VGG16 Model

We use the pre-trained model's architecture to create a new dataset from our input images in this approach. We'll import the Convolutional and...