SYMBOL INDEX (22 symbols across 5 files)

FILE: adamw.py
  class AdamW (line 5) | class AdamW(Optimizer):
    method __init__ (line 6) | def __init__(self, params, lr=1e-4, betas=(0.9, 0.999), eps=1e-5, weig...
    method step (line 13) | def step(self, closure=None):

FILE: autoencoder.py
  class GELU (line 34) | class GELU(nn.Module):
    method forward (line 35) | def forward(self, x):
  class VAE (line 38) | class VAE(nn.Module):
    method __init__ (line 39) | def __init__(self):
    method encode (line 49) | def encode(self, x):
    method reparametrize (line 53) | def reparametrize(self, mu, logvar):
    method decode (line 61) | def decode(self, z):
    method forward (line 65) | def forward(self, x):
  function loss_function (line 74) | def loss_function(recon_x, x, mu, logvar):

FILE: example_cpu.py
  function calculate_reward (line 20) | def calculate_reward(model, loader, index, risk = True, policy = None, p...
  function worker (line 61) | def worker(q_policies, q_results, calc_reward_args):
  function train (line 80) | def train(model, index, risk = True):
  class ARSModel (line 130) | class ARSModel():
    method __init__ (line 131) | def __init__(self, numFeatures, NumChannels, preload):
    method genDeltas (line 140) | def genDeltas(self):
    method forward (line 144) | def forward(self, state, policy = None):

FILE: example_gpu.py
  function calculate_reward (line 24) | def calculate_reward(model, loader, index, risk = 1.0, skip = None):
  function train (line 75) | def train(model, optimizer, index, risk = 1.0):

FILE: shared.py
  function plot_function (line 59) | def plot_function(epoch_weights):