gasiort

# most of the code is taken form Practical RL week0 assignment (https://github.com/yandexdataschool/Practical_RL/tree/master/week0)

# I'm just trying out OpenAI submission interface

import numpy as np
import os

import gym
from gym import wrappers

gasiort

class GuidedBackprop(torch.nn.Module):
    def __init__(self, model, n_classes):
        super().__init__()
        self.model = model
        self.n_classes = n_classes
        self.gradients = None
        self.model.eval()
        self.update_relus()
        self.hook_first_layer()