Pretrained model file not found

Accelerated Computing Intelligent Video Analytics TAO Toolkit
1

The error i got:

AssertionError: Pretrained model file not found: /workspace/pretrained_model/tlt_resnet18_detectnet_v2_v1/

I used below configurattion

model_config {
  pretrained_model_file: "/workspace/pretrained_model/tlt_resnet18_detectnet_v2_v1/"
  num_layers: 18
  freeze_blocks: 0
  arch: "resnet"
  use_batch_norm: true
  objective_set: {
    cov {}
    bbox {
      scale: 35.0
      offset: 0.5
    }
  }
  training_precision {
    backend_floatx: FLOAT32
 }
}

I also tried altering the model path like:

pretrained_model_file: "/workspace/pretrained_model"
pretrained_model_file: "/workspace/pretrained_model/*"
pretrained_model_file: "/workspace/pretrained_model/tlt_resnet18_detectnet_v2_v1"
pretrained_model_file: "/workspace/pretrained_model/tlt_resnet18_detectnet_v2_v1/"

But nothing seems to work, i also checked the pretrained_model directory this file “tlt_resnet18_detectnet_v2_v1” is present there.

2

Hi mneonizer
Can you run below command and paste the result?
$ ll /workspace/pretrained_model/tlt_resnet18_detectnet_v2_v1/

If there is a resnet18.hdf5 file, please modify the spec file as below.

pretrained_model_file: "/workspace/pretrained_model/tlt_resnet18_detectnet_v2_v1/resnet18.hdf5"
3

Here is the output:

root@cef161d4132d:/workspace# ll /workspace/pretrained_model/tlt_resnet18_detectnet_v2_v1/
total 91176
drwx------ 2 root root     4096 Nov 12 05:51 ./
drwxr-xr-x 3 root root     4096 Nov 12 05:52 ../
-rw------- 1 root root      253 Nov 12 05:51 license.txt
-rw------- 1 root root 93345248 Nov 12 05:52 resnet18.hdf5