wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-07-23 13:10:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

added MAF option to model

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-01-17 11:42:46 +01:00
parent 744eb22ec4
commit 04bdd7bbf2
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/tempflow/tempflow_network.py
+2 -2
View File
@@ -5,7 +5,7 @@ import torch.nn as nn
import numpy as np
from pts.modules import RealNVP, FlowOutput, MeanScaler, NOPScaler
from pts.modules import RealNVP, MAF, FlowOutput, MeanScaler, NOPScaler
from pts.model import weighted_average
@@ -54,7 +54,7 @@ class TempFlowTrainingNetwork(nn.Module):
batch_first=True,
)
flow_cls = {"RealNVP": RealNVP}[flow_type]
flow_cls = {"RealNVP": RealNVP, "MAF": MAF,}[flow_type]
self.flow = flow_cls(
input_size=target_dim,
n_blocks=n_blocks,