ZolotukhinM

#!/usr/bin/python
import random

color = {
    "BOLD": "\033[1m",
    "UNDERLINE": "\033[4m",
    "GREEN": "\033[92m",
    "RED": "\033[91m",
    "BLUE": "\033[94m",
    "CLEAR": "\033[0m",

ZolotukhinM

import torch
import os
from timeit import default_timer as timer

class DeepAndWide(torch.nn.Module):
    def __init__(self, num_features=50):
        super(DeepAndWide, self).__init__()
        self.mu = torch.randn(1, num_features)
        self.sigma = torch.randn(1, num_features)
        self.fc_w = torch.randn(1, num_features + 1)

ZolotukhinM

import torch
import os
from timeit import default_timer as timer

class DeepAndWide(torch.nn.Module):
    def __init__(self, num_features=50):
        super(DeepAndWide, self).__init__()
        self.mu = torch.rand(1, num_features)
        self.sigma = torch.rand(1, num_features)
        self.fc_w = torch.rand(1, num_features + 1)