ShoufaChen

def summarize_tensor(x):
    return f"\033[34m{str(tuple(x.shape)).ljust(24)}\033[0m (\033[31mmin {x.min().item():+.4f}\033[0m / \033[32mmean {x.mean().item():+.4f}\033[0m / \033[33mmax {x.max().item():+.4f}\033[0m)"


class ModelActivationPrinter:
    def __init__(self, module, submodules_to_log):
        self.id_to_name = {
            id(module): str(name) for name, module in module.named_modules()
        }
        self.submodules = submodules_to_log

ShoufaChen

# Modified by Shoufa Chen,

import math
import random
import copy
import torch
import torch.nn as nn
import torch.nn.functional as F
from fvcore.nn import sigmoid_focal_loss_jit
from slowfast.models.losses import focal_loss_wo_logits_jit

ShoufaChen

"""
A PyTorch version of `botnet`.
Original TensorFlow version: https://gist.github.com/aravindsrinivas/56359b79f0ce4449bcb04ab4b56a57a2
"""

import torch
import torch.nn as nn
import torch.nn.functional as F

BATCH_NORM_DECAY = 1 - 0.9  # pytorch batch norm `momentum = 1 - counterpart` of tensorflow

ShoufaChen

1 How can I uninstall a nvidia driver completely ? 完全卸载驱动

• Solution 1

sudo apt-get remove --purge nvidia-*

• solution 2

sudo apt-get remove --purge '^nvidia-.*'