File size: 198 Bytes
32b2aaa
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
from dataclasses import dataclass

from ..hparams import HParams as HParamsBase


@dataclass(frozen=True)
class HParams(HParamsBase):
    batch_size_per_gpu: int = 128
    distort_prob: float = 0.5