Skip to content

Commit

Permalink
updates
Browse files Browse the repository at this point in the history
  • Loading branch information
Anirudh Goyal committed Aug 19, 2020
1 parent 4df3e3e commit 20407c8
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions event_based/experiment_wikitext2.sh
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,8 @@ name="/home/anirudh/icml_blocks/sparse_relational/Blocks_wiki2/experimental_Bloc
name="${name//./}"
echo Running version $name

#Replace d_k=32, d_v=32 https://github.com/anirudh9119/RIMs/blob/master/event_based/blocks_core.py#L47
# self.att_out = 510. (This should really be a hyperparameter.)

#./experiment_wikitext2.sh 0.0007 510 510 5 5 0.5. It should get a test ~ 102.

Expand All @@ -25,4 +27,6 @@ python /home/anirudh/icml_blocks/sparse_relational/train_wiki.py --tied --cuda -


#To run 2 layered RIM baseline.
#./experiment_wikitext2.sh 0.0007 510 510 5 5 0.5. It should get a test ~ 99.

#python /home/anirudh/icml_blocks/sparse_relational/train_wiki.py --tied --cuda --cudnn --algo blocks --data /home/anirudh/icml_blocks/sparse_relational/data/wikitext-2 --name $name --lr $lr --drop $drop --nhid $dim1 $dim1 --num_blocks $block1 $block1 --topk $topk1 $topk1 --use_inactive --nlayers 2 --emsize $em --log-interval $log

0 comments on commit 20407c8

Please sign in to comment.