Correctly initializing the logit scale parameter

adding numpy import
This commit is contained in:
Jong Wook Kim 2021-03-22 22:09:57 -04:00 committed by GitHub
parent 43c953e231
commit 290ac5cb15
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions

View File

@ -1,6 +1,7 @@
from collections import OrderedDict
from typing import Tuple, Union
import numpy as np
import torch
import torch.nn.functional as F
from torch import nn