small correction about argument parsing in copy_task.py

[softgearko]

When I run 'python3 copy_task.py --rnn_num_layers 3 --rnn_size 64 --max_seq_length 10 --memory_size 20 --memory_vector_dim 8 --vector_dim 4' , I got an error about argument converting.
So, I added "type=int," or "type=float" into add_argument()s in copy_task.py.

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--mode', default="train")
    parser.add_argument('--restore_training', default=False)
    parser.add_argument('--test_seq_length', type=int, default=20)
    parser.add_argument('--model', default="NTM")
    parser.add_argument('--rnn_size', type=int, default=128)
    parser.add_argument('--rnn_num_layers', type=int, default=3)
    parser.add_argument('--max_seq_length', type=int, default=15)
    parser.add_argument('--memory_size', type=int, default=128)
    parser.add_argument('--memory_vector_dim', type=int, default=20)
    parser.add_argument('--batch_size', type=int, default=10)
    parser.add_argument('--vector_dim', type=int, default=8)
    parser.add_argument('--shift_range', type=int, default=1)
    parser.add_argument('--num_epoches', type=int, default=1000000)
    parser.add_argument('--learning_rate', type=float, default=1e-4)
    parser.add_argument('--save_dir', default='./save/copy_task')
    parser.add_argument('--tensorboard_dir', default='./summary/copy_task')
    args = parser.parse_args()
    if args.mode == 'train':
        train(args)
    elif args.mode == 'test':
        test(args)

I want to see a result figure at in your slide page 11. What arguments do I run copy_task.py with? Are the parameters above right?

[snowkylin]

If you just want to replicate the result in page 11 of my slide, you can set parameters smaller. memory_size and memory_vector_dim can be set to 20 and 8.