smsharma

smsharma

smsharma

import numpy as np
from torch.utils.data import Dataset

def load_and_check(filename, memmap=False):
    """ Load numpy array optionally with memmap
    """
    if memmap:
        data = np.load(filename, mmap_mode="c")
    else:
        data = np.load(filename)

smsharma

smsharma

smsharma

smsharma

import sys
sys.path.append("../")

from itertools import permutations

import torch
from torch import nn
from torch.distributions import Normal
from torch.nn import CrossEntropyLoss
import pytorch_lightning as pl

smsharma

smsharma

smsharma