`popgym.util.benchmark`

Module Contents

Functions

`main`()
`train_closure`(model, device)
`inference_closure`(model, device)

Attributes

`BATCH`
`TIME`
`DIM`
`h`
`SAMPLES`

popgym.util.benchmark.BATCH = 64

popgym.util.benchmark.TIME = 1024

popgym.util.benchmark.DIM = 256

popgym.util.benchmark.h = 128

popgym.util.benchmark.SAMPLES = 10

popgym.util.benchmark.main()

popgym.util.benchmark.train_closure(model, device)

popgym.util.benchmark.inference_closure(model, device)