Method default_config

axlearn/common/state_builder_test.py:454–458 · view source on GitHub ↗

(cls)

Source from the content-addressed store, hash-verified

452
453	@classmethod
454	def default_config(cls):
455	cfg = super().default_config()
456	cfg.layer = Linear.default_config().set(input_dim=2, output_dim=3)
457	cfg.num_layers = 2
458	return cfg
459
460
461	class DummyModel(BaseModel): # pylint: disable=abstract-method

test_converter_string_and_enum_argsMethod · 0.45

test_converter_defaultMethod · 0.45

test_chainMethod · 0.45

_test_get_builderMethod · 0.45

test_simple_chain_builderMethod · 0.45

test_chain_builder_errorMethod · 0.45

_mock_bert_trainer_configMethod · 0.45

test_truncationMethod · 0.45

test_truncation_target_longer_than_sourceMethod · 0.45

test_truncation_incompatible_shapeMethod · 0.45

test_replace_target_prefix_with_sourceMethod · 0.45

test_keep_targetMethod · 0.45

setMethod · 0.45

no test coverage detected