File size: 720 Bytes
0df76ba |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import json
import transformers
class ContextualModelConfig(transformers.configuration_utils.PretrainedConfig):
"""We create a dummy configuration class that will just set properties
based on whatever kwargs we pass in.
When this class is initialized (see experiments.py) we pass in the
union of all data, model, and training args, all of which should
get saved to the config json.
"""
def __init__(self, **kwargs):
for key, value in kwargs.items():
try:
json.dumps(value)
setattr(self, key, value)
except TypeError:
# value was not JSON-serializable, skip
continue
super().__init__()
|