Correctly initializing the logit scale parameter
adding numpy import
This commit is contained in:
parent
43c953e231
commit
290ac5cb15
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue