def get_architecture(checkpoint_args, args):
	"""
	Loads architecure-specific arguments from a checkpoint

	Args
		checkpoint_args: the arguments of the checkpoint
		args: the arguments of this run
	Returns
		args: the updated arguments of this run
	"""

	args.num_layers_shared_enc = checkpoint_args.num_layers_shared_enc
	args.num_layers_passage_enc = checkpoint_args.num_layers_passage_enc
	args.num_layers_question_enc = checkpoint_args.num_layers_question_enc
	args.num_layers_qa_enc = checkpoint_args.num_layers_qa_enc
	args.num_layers_dec = checkpoint_args.num_layers_dec
	args.d_model = checkpoint_args.d_model
	args.d_inner = checkpoint_args.d_inner
	args.heads = checkpoint_args.heads
	args.coattention = checkpoint_args.coattention

	return args