SYMBOL INDEX (70 symbols across 6 files)

FILE: cl_training.py
  function get_reward (line 43) | def get_reward(preds, targets, policy):
  function train (line 58) | def train(epoch):
  function test (line 136) | def test(epoch):

FILE: finetune.py
  function get_reward (line 42) | def get_reward(preds, targets, policy):
  function train (line 57) | def train(epoch):
  function test (line 117) | def test(epoch):

FILE: models/base.py
  class Identity (line 15) | class Identity(nn.Module):
    method __init__ (line 16) | def __init__(self):
    method forward (line 18) | def forward(self, x):
  class Flatten (line 21) | class Flatten(nn.Module):
    method __init__ (line 22) | def __init__(self):
    method forward (line 24) | def forward(self, x):
  function conv3x3 (line 27) | def conv3x3(in_planes, out_planes, stride=1):
  class BasicBlock (line 31) | class BasicBlock(nn.Module):
    method __init__ (line 34) | def __init__(self, inplanes, planes, stride=1):
    method forward (line 41) | def forward(self, x):
  class Bottleneck (line 52) | class Bottleneck(nn.Module):
    method __init__ (line 55) | def __init__(self, inplanes, planes, stride=1):
    method forward (line 65) | def forward(self, x):
  class DownsampleB (line 81) | class DownsampleB(nn.Module):
    method __init__ (line 83) | def __init__(self, nIn, nOut, stride):
    method forward (line 88) | def forward(self, x):

FILE: models/resnet.py
  class FlatResNet (line 19) | class FlatResNet(nn.Module):
    method seed (line 21) | def seed(self, x):
    method forward (line 28) | def forward(self, x, policy):
    method forward_single (line 56) | def forward_single(self, x, policy):
    method forward_full (line 76) | def forward_full(self, x):
  class FlatResNet32 (line 92) | class FlatResNet32(FlatResNet):
    method __init__ (line 94) | def __init__(self, block, layers, num_classes=10):
    method seed (line 126) | def seed(self, x):
    method _make_layer (line 130) | def _make_layer(self, block, planes, blocks, stride=1):
  class FlatResNet224 (line 145) | class FlatResNet224(FlatResNet):
    method __init__ (line 147) | def __init__(self, block, layers, num_classes=1000):
    method seed (line 179) | def seed(self, x):
    method _make_layer (line 183) | def _make_layer(self, block, planes, blocks, stride=1):
  class Policy32 (line 205) | class Policy32(nn.Module):
    method __init__ (line 207) | def __init__(self, layer_config=[1,1,1], num_blocks=15):
    method load_state_dict (line 216) | def load_state_dict(self, state_dict):
    method forward (line 222) | def forward(self, x):
  class Policy224 (line 229) | class Policy224(nn.Module):
    method __init__ (line 231) | def __init__(self, layer_config=[1,1,1,1], num_blocks=16):
    method load_state_dict (line 247) | def load_state_dict(self, state_dict):
    method forward (line 252) | def forward(self, x):
  class StepResnet32 (line 261) | class StepResnet32(FlatResNet32):
    method __init__ (line 263) | def __init__(self, block, layers, num_classes, joint=False):
    method seed (line 276) | def seed(self, x):
    method step (line 284) | def step(self, action):
    method step_single (line 298) | def step_single(self, action):
    method predict (line 313) | def predict(self):
  class StepPolicy32 (line 320) | class StepPolicy32(nn.Module):
    method __init__ (line 322) | def __init__(self, layer_config):
    method forward (line 328) | def forward(self, state):

FILE: test.py
  class FConv2d (line 20) | class FConv2d(nn.Conv2d):
    method __init__ (line 22) | def __init__(self, in_channels, out_channels, kernel_size, stride=1,
    method forward (line 28) | def forward(self, x):
  class FLinear (line 35) | class FLinear(nn.Linear):
    method __init__ (line 36) | def __init__(self, in_features, out_features, bias=True):
    method forward (line 40) | def forward(self, x):
  function count_flops (line 45) | def count_flops(model, reset=True):
  function test (line 61) | def test():

FILE: utils.py
  function save_args (line 11) | def save_args(__file__, args):
  function performance_stats (line 16) | def performance_stats(policies, rewards, matches):
  class LrScheduler (line 31) | class LrScheduler:
    method __init__ (line 32) | def __init__(self, optimizer, base_lr, lr_decay_ratio, epoch_step):
    method adjust_learning_rate (line 38) | def adjust_learning_rate(self, epoch):
  function load_weights_to_flatresnet (line 49) | def load_weights_to_flatresnet(source_model, target_model):
  function load_checkpoint (line 85) | def load_checkpoint(rnet, agent, load):
  function get_transforms (line 103) | def get_transforms(rnet, dset):
  function get_dataset (line 154) | def get_dataset(model, root='data/'):
  function get_model (line 172) | def get_model(model):