Skip to content

Instantly share code, notes, and snippets.

@chetanambi

chetanambi/reddit_tifu_long.py

Created August 4, 2020 15:45
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save chetanambi/cfac43e917a3f2c866f97fac15033ae3 to your computer and use it in GitHub Desktop.
Save chetanambi/cfac43e917a3f2c866f97fac15033ae3 to your computer and use it in GitHub Desktop.
Download ZIP
reddit_tifu_long
Raw
reddit_tifu_long.py
save_path = "tfrecord:/content/pegasus/pegasus/data/testdata/test_pattern_1.tfrecord"
@registry.register("reddit_tifu_long_transformer")
def reddit_tifu_long_transformer(param_overrides):
return transformer_params(
{
"train_pattern": save_path,
"dev_pattern": save_path,
"test_pattern": save_path,
"max_input_len": 1024,
"max_output_len": 128,
"train_steps": 8000,
"learning_rate": 0.0001,
"batch_size": 8,
}, param_overrides)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment