| import argparse | |
| from tqdm import tqdm | |
| import torch | |
| import torch.nn.functional as F | |
| from torchvision import models, datasets, transforms | |
| def get_CIFAR10(root="./"): |
| # https://github.com/samiraabnar/attention_flow | |
| # https://github.com/google-research/vision_transformer/issues/27 | |
| # https://github.com/google-research/vision_transformer/issues/18 | |
| # https://github.com/faustomorales/vit-keras/blob/65724adcfd3979067ce24734f08df0afa745637d/vit_keras/visualize.py#L7-L45 | |
| # https://github.com/jeonsworld/ViT-pytorch/blob/main/visualize_attention_map.ipynb | |
| import torch | |
| import numpy as np | |
| import matplotlib.pyplot as plt | |
| from PIL import Image, ImageFilter |
| import torch | |
| import torch.nn as nn | |
| import torch.nn.functional as F | |
| import torchvision.models as tmodels | |
| from functools import partial | |
| import collections | |
| # dummy data: 10 batches of images with batch size 16 | |
| dataset = [torch.rand(16,3,224,224).cuda() for _ in range(10)] |
The repository for the assignment is public and Github does not allow the creation of private forks for public repositories.
The correct way of creating a private frok by duplicating the repo is documented here.
For this assignment the commands are:
git clone --bare git@github.com:usi-systems/easytrace.git
