kd5678
/

prefpaintReward

Model card Files Files and versions Community

prefpaintReward / configs.yaml

kd5678's picture

Update configs.yaml

eff510e verified 5 months ago

200 Bytes

	Reward: !munch.Munch
	mlp_dim: 1024
	model_path: ./prefpaintReward.pt
	ViT: !munch.Munch
	dropout: 0.1
	feature_dim: 1024
	mlp_dim: 1024
	num_heads: 16

	fix_base: false
	fix_rate: 0.7
	group: False