hiranoko

# vgg.py
import torch
import torch.nn as nn
from collections import OrderedDict

_cfgs = {
    #  M = MaxPool, 64 = conv out-channels
    "A":  [64, "M", 128, "M", 256, 256, "M", 512, 512, "M", 512, 512, "M"],            # VGG-11
    "B":  [64, 64, "M", 128, 128, "M", 256, 256, "M", 512, 512, "M", 512, 512, "M"],    # VGG-13
    "D":  [64, 64, "M", 128, 128, "M", 256, 256, 256, "M", 512, 512, 512, "M",

hiranoko

#!/usr/bin/env python3
# pl_main.py
import argparse
import torch
from torch.utils.data import DataLoader
import torchvision.transforms as T
import torchvision.datasets as dsets
import torchvision.models as models
import pytorch_lightning as pl
from pytorch_lightning.callbacks import ModelCheckpoint, LearningRateMonitor

hiranoko

#　インデックス検索

import numpy as np
import torch
import torch.nn.functional as F
from sklearn.neighbors import NearestNeighbors

# パラメータ