Skip to content

AshvinAN/nvidia-model-deployment

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

11 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

NVIDIA Model Deployment – TensorFlow Example

This project demonstrates deploying a simple TensorFlow model using the NVIDIA Deep Learning Institute (DLI) lab environment.

πŸš€ Project Highlights

  • Built and trained a basic TensorFlow model using Keras Sequential API.
  • Saved the trained model in SavedModel format under models/simple-tensorflow-model/1.
  • Ran inference using dummy input data and verified model predictions.
  • Organized and uploaded the model files to GitHub for showcasing.

πŸ—‚οΈ Folder Structure

models/ └── simple-tensorflow-model/ └── 1/ β”œβ”€β”€ Screenshots/ β”œβ”€β”€ variables/ β”‚ β”œβ”€β”€ variables.data-00000-of-00001 β”‚ └── variables.index └── saved_model.pb

πŸ“Έ Screenshots

  • βœ… Model training & saving
  • βœ… Folder structure
  • βœ… Inference output
  • βœ… GitHub hosted model repo

🧠 Skills Practiced

  • Model creation with TensorFlow/Keras
  • SavedModel format deployment
  • Git/GitHub project management
  • Hands-on with model inference