popgym.util.benchmark
=====================

.. py:module:: popgym.util.benchmark


Attributes
----------

.. autoapisummary::

   popgym.util.benchmark.BATCH
   popgym.util.benchmark.TIME
   popgym.util.benchmark.DIM
   popgym.util.benchmark.h
   popgym.util.benchmark.SAMPLES


Functions
---------

.. autoapisummary::

   popgym.util.benchmark.main
   popgym.util.benchmark.train_closure
   popgym.util.benchmark.inference_closure


Module Contents
---------------

.. py:data:: BATCH
   :value: 64


.. py:data:: TIME
   :value: 1024


.. py:data:: DIM
   :value: 256


.. py:data:: h
   :value: 128


.. py:data:: SAMPLES
   :value: 10


.. py:function:: main()

.. py:function:: train_closure(model, device)

.. py:function:: inference_closure(model, device)