Fix output directory in LaunchLocalTrainingJob.ipynb

Author: mitchellspryn

AirSimDistributedRL/LaunchLocalTrainingJob.ipynb

@@ -41,7 +41,9 @@
  {
  "cell_type": "code",
  "execution_count": 4,
- "metadata": {},
+ "metadata": {
+ "collapsed": true
+ },
  "outputs": [],
  "source": [
  "batch_update_frequency = 300\n",
@@ -60,18 +62,16 @@
  "cell_type": "markdown",
  "metadata": {},
  "source": [
- "We will now generate a training batch file. Run this file from an activated python environment to train."
+ "We will now generate a training batch file. The file will be written to Share\\scripts_downpour\\app. Run this file from an activated python environment in that directory to train."
  ]
  },
  {
  "cell_type": "code",
- "execution_count": 5,
- "metadata": {
- "collapsed": true
- },
+ "execution_count": 11,
+ "metadata": {},
  "outputs": [],
  "source": [
- "train_cmd = 'python Share\\\\app\\\\scripts_downpour\\\\distributed_agent.py'\n",
+ "train_cmd = 'python distributed_agent.py'\n",
  "train_cmd += ' batch_update_frequency={0}'.format(batch_update_frequency)\n",
  "train_cmd += ' max_epoch_runtime_sec={0}'.format(max_epoch_runtime_sec)\n",
  "train_cmd += ' per_iter_epsilon_reduction={0}'.format(per_iter_epsilon_reduction)\n",
@@ -83,15 +83,15 @@
  "train_cmd += ' airsim_path={0}'.format(airsim_path)\n",
  "train_cmd += ' data_dir={0}'.format(data_dir)\n",
  "\n",
- "with open('train.bat', 'w') as f:\n",
+ "with open(os.path.join(os.getcwd(), 'Share/scripts_downpour/app/train.bat'), 'w') as f:\n",
  " f.write(train_cmd)"
  ]
  },
  {
  "cell_type": "markdown",
  "metadata": {},
  "source": [
- "Note that training the m"
+ "Note that training the model from scratch can take up to 5 days on a powerful GPU. Using pretrained weights, the model should begin to visibly converge after 3 hours of training on a GTX 1080."
  ]
  }
  ],