Skip to content
Snippets Groups Projects

WIP: Generic trainer

Closed Anjith GEORGE requested to merge generic_trainer into master
2 files
+ 2
1
Compare changes
  • Side-by-side
  • Inline
Files
2
import torch
from torch import nn
from torchvision import models
import numpy as np
class MCDeepPixBiS(nn.Module):
Loading