Unet tlt model files vs checkpoint tlt files after X epochs

I am training a tao 5.1 unet model with the following command:

!tao model unet train --gpus $NUM_GPUS \
                      --gpu_index $GPU_INDEX \
                      -e $SPECS_DIR/unet_train_vgg_6S250.txt \
                      -r $USER_EXPERIMENT_DIR/unpruned \
                      -m $USER_EXPERIMENT_DIR/pretrained_vgg16/pretrained_semantic_segmentation_vvgg16/vgg_16.hdf5 \
                      -n 6SBan003

The spec file specifies 250 epochs.

The final model will be at unpruned/weights/6SBan003.tlt

Also, there is a file at unpruned/model.epoch-250.tlt

Are they the same? Can I use model.epoch-250.tlt as input to the prune operation, or export as is to a
tensorrt engine?

Thanks!

David

2 posts - 2 participants

Read full topic