You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

config 1.2 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. [test_seq_label_trainer]
  2. epochs = 1
  3. batch_size = 32
  4. validate = true
  5. save_best_dev = true
  6. use_cuda = true
  7. [test_seq_label_model]
  8. rnn_hidden_units = 100
  9. rnn_layers = 1
  10. rnn_bi_direction = true
  11. word_emb_dim = 100
  12. dropout = 0.5
  13. use_crf = true
  14. use_cuda = true
  15. [test_seq_label_tester]
  16. save_output = true
  17. validate_in_training = true
  18. save_dev_input = false
  19. save_loss = true
  20. batch_size = 1
  21. rnn_hidden_units = 100
  22. rnn_layers = 1
  23. rnn_bi_direction = true
  24. word_emb_dim = 100
  25. dropout = 0.5
  26. use_crf = true
  27. use_cuda = true
  28. [POS_infer]
  29. rnn_hidden_units = 100
  30. rnn_layers = 1
  31. rnn_bi_direction = true
  32. word_emb_dim = 100
  33. vocab_size = 53
  34. num_classes = 27
  35. [text_class]
  36. epochs = 1
  37. batch_size = 10
  38. validate = false
  39. save_best_dev = false
  40. use_cuda = true
  41. learn_rate = 1e-3
  42. momentum = 0.9
  43. model_name = "class_model.pkl"
  44. [snli_trainer]
  45. epochs = 5
  46. batch_size = 32
  47. validate = true
  48. save_best_dev = true
  49. use_cuda = true
  50. learn_rate = 1e-4
  51. loss = "cross_entropy"
  52. print_every_step = 1000
  53. [snli_tester]
  54. batch_size = 512
  55. use_cuda = true
  56. [snli_model]
  57. model_name = "snli_model.pkl"
  58. embed_dim = 300
  59. hidden_size = 300
  60. batch_first = true
  61. dropout = 0.5
  62. gpu = true
  63. embed_file = "./../data_for_tests/glove.840B.300d.txt"
  64. embed_pkl = "./snli/embed.pkl"
  65. examples = 0