Flag Demo

Usage of ./flag:
  -bs int
        batch size (default 32)
  -criterion string
        criterion (default "mse")
  -dropout float
        dropout probability (default 0.2)
  -epochs int
        epochs (default 10)
  -optimizer string
        optimizer (default "adam")