Please provide the following information when requesting support.
• Hardware (NVIDIA RTX 3080Ti)
• Network Type (Efficientdet)
• TLT Version (tao-tf1)
• Training Spec file
training_config {
train_batch_size: 8
iterations_per_loop: 1
use_xla: False
disable_logging: False
stop_at_epoch: 2
resume: False
checkpoint_period: 10
lr_warmup_init: 0.00005
learning_rate: 0.1
keep_checkpoint_max: 10
num_examples_per_epoch: 50
num_epochs: 2
profile_skip_steps: 100
tf_random_seed: 42
}
dataset_config {
num_classes: 91
image_size: “512, 512”
training_file_pattern: “/workspace/tao-tf1/nvidia_tao_tf1/cv/dataset/coco/tfrecords/"
validation_file_pattern: "/workspace/tao-tf1/nvidia_tao_tf1/cv/dataset/coco/tfrecords/”
validation_json_file: “/workspace/tao-tf1/nvidia_tao_tf1/cv/dataset/coco/annotations/instances_val2017.json”
}
eval_config {
eval_batch_size: 8
eval_epoch_cycle: 1
eval_after_training: False
eval_samples: 5
}
augmentation_config {
random_crop_min_scale: 0.1
random_crop_min_scale: 2.0
rand_hflip: True
}
model_config {
model_name: ‘efficientdet-d0’
}
• How to reproduce the issue ?
python efficientdet/scripts/prune.py -m /workspace/tao-tf1/nvidia_tao_tf1/cv/efficientdet/results/model.epoch-0.tlt -o /workspace/tao-tf1/nvidia_tao_tf1/cv/efficientdet/results/prune -pth 0.5
INFO:tensorflow:Restoring parameters from /tmp/tmpuufymj1m/model.ckpt-1837
2024-07-05 04:59:04,389 [TAO Toolkit] [INFO] tensorflow 1284: Restoring parameters from /tmp/tmpuufymj1m/model.ckpt-1837
2024-07-05 04:59:04,886 [TAO Toolkit] [INFO] main 227: Pruning process will take some time. Please wait…
2024-07-05 04:59:04,918 [TAO Toolkit] [INFO] nvidia_tao_tf1.core.pruning.pruning 981: Exploring graph for retainable indices
2024-07-05 04:59:04,918 [TAO Toolkit] [INFO] root 2102: Unknown layer type: <class ‘tensorflow.python.keras.engine.input_layer.InputLayer’>
Traceback (most recent call last):
File “efficientdet/scripts/prune.py”, line 310, in
main()
File “efficientdet/scripts/prune.py”, line 306, in main
raise e
File “efficientdet/scripts/prune.py”, line 294, in main
run_pruning(args)
File “efficientdet/scripts/prune.py”, line 229, in run_pruning
pruned_model = prune(
File “/workspace/tao-tf1/nvidia_tao_tf1/core/pruning/pruning.py”, line 1606, in prune
return pruner.prune(model, layer_config_overrides, output_layers_with_outbound_nodes)
File “/workspace/tao-tf1/nvidia_tao_tf1/core/pruning/pruning.py”, line 1211, in prune
model = self._explore(model)
File “/workspace/tao-tf1/nvidia_tao_tf1/core/pruning/pruning.py”, line 1142, in _explore
raise NotImplementedError(“Unknown layer type: %s” % type(layer))
NotImplementedError: Unknown layer type: <class ‘tensorflow.python.keras.engine.input_layer.InputLayer’>
2 posts - 2 participants