lie_conv/datasets.py
File datasets.py
has 512 lines of code (exceeds 250 allowed). Consider refactoring. Open
Open
import math
import torch
import torchvision.transforms as transforms
import torch.nn as nn
import torch.nn.functional as F
Function __init__
has 8 arguments (exceeds 4 allowed). Consider refactoring. Open
Open
def __init__(self, root_dir=default_root_dir, train=True, download=True, n_systems=100, regen=False,
Function __init__
has 7 arguments (exceeds 4 allowed). Consider refactoring. Open
Open
def __init__(self, root_dir=default_root_dir, train=True, download=True, n_systems=100, space_dim=2, regen=False,