/tinytorchtrainer

A small framework to train image classifiers in PyTorch

Primary LanguagePythonCreative Commons Attribution Share Alike 4.0 InternationalCC-BY-SA-4.0

tinytorchtrainer

pytest CC BY-SA 4.0

Successor of pytorch-pretrained-cnns minus pytorch lightning

Scripts:

  • train.py: Training with various parameters
  • eval.py: Runs evaluation of a given checkpoint with validation dataset
  • eval_adv_rob.py: Runs evaluation of a given checkpoint with validation dataset under adversarial attacks of AutoAttack
  • plot_convolutions.py: Plots the 3x3 conv filters of a given checkpoint
  • plot_feature_maps.py: Plots the feature maps of selected layers of a given checkpoints for a given input sample
  • init_data.py: Prepares the data independent of training (usefull before sweeps)
  • ckpt_info.py: Dumps info stored in a checkpoint
  • to_onnx.py: Exports checkpoints to ONNX

Credits

Code partially taken from https://github.com/huyvnphan/PyTorch_CIFAR10, https://github.com/1M50RRY/resnet18-preact, https://github.com/pytorch/vision/, https://github.com/andreasveit/densenet-pytorch/.