Skip to content

My reproduction of Google's im2txt (or Show and Tell) model using scripts for conducting training and evaluation on a SLURM cluster.

License

Notifications You must be signed in to change notification settings

taylorpaul/ShowAndTell-SLURM

Repository files navigation

ShowAndTell-SLURM

My reproduction of Google's im2txt (or Show and Tell) model using scripts for conducting training and evaluation on a SLURM cluster.

Getting Started:

I followed this tutorial from the im2text repository. In order to develop the scripts included in this repository for training the model using SLURM. Start there and run the appropriate script. Each script has TODO's requiring updates for your directories, etc. Here are the basic steps to reproduce the training once all dependecies are installed(Need help installing TF on CENTOS 6.8 without root? Check here):

  1. Clone the im2txt model into your desired directory
  2. Download and process the MSCOCO data using MSCOCO.sh
  3. Download the InceptionV3 checkpoint using get_inception.sh
  4. Modify tensorflowShowandTell0.sh and tensorflowShowandTell1.sh for your system.
  5. Initialize the training and evaluation using im2txttrain.sh to call the scripts modified in step 4
  6. After 5,000 steps complete, run tensorboard.sh to start a tensorboard server.
  7. Follow this gist to access the server.
  8. You can modify and run showAndTellClassify.sh to test your model when the script from step 5 completes.
  9. Run trainWithInception.sh to complete an additional 2 Million training steps including updates to the Inception network.

About

My reproduction of Google's im2txt (or Show and Tell) model using scripts for conducting training and evaluation on a SLURM cluster.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages