SAM-DiffSR / sam_diffsr /configs /rrdb /df2k4x_pretrain.yaml
Traly's picture
init
193c713
raw
history blame contribute delete
229 Bytes
base_config:
- ../sr_base.yaml
- ../df2k4x.yaml
trainer_cls: tasks.rrdb.RRDBDf2kTask
# model
hidden_size: 64
lr: 0.0002
num_block: 17
# train and eval
max_updates: 100000
batch_size: 64
eval_batch_size: 1
valid_steps: 3