Skip to content

Predictor Net Updates

Pre-release
Pre-release
Compare
Choose a tag to compare
@cpaxton cpaxton released this 07 Nov 18:47
· 4454 commits to master since this release
9b41b66

This release contains a preliminary version of our proposed predictor network architecture for predictive modeling of task structure. You can train such a model end to end via a command such as:

rosrun costar_models ctp_model_tool \
  --data_file rpy.npz \
  --model predictor \
  -e 1000 \
  --features multi \
  --batch_size 24  \
  --optimizer adam \
  --lr 0.001 \
  --upsampling conv_transpose \
  --use_noise true \
  --noise_dim 32  \
  --steps_per_epoch 300 \
  --dropout_rate 0.2 \
  --skip_connections 1

Which constructs an end-to-end trainable version of the network that can predict 4 hypotheses.