Models package

Package for (PyTorch) models of all types.

Models.erfh5_ConvModel

class Models.erfh5_ConvModel.DrySpotModel

Bases: sphinx.ext.autodoc.importer._MockObject

forward(x)

class Models.erfh5_ConvModel.S20Channel4toDrySpot

Bases: sphinx.ext.autodoc.importer._MockObject

forward(_inputs)

class Models.erfh5_ConvModel.S20DeconvToDrySpotEff(pretrained='', checkpoint_path=None, freeze_nlayers=0)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

class Models.erfh5_ConvModel.S20DeconvToDrySpotEff2(pretrained='', checkpoint_path=None, freeze_nlayers=0, round_at: float = None, demo_mode=False)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

class Models.erfh5_ConvModel.S20DeconvToDrySpotEff3(pretrained='', checkpoint_path=None, freeze_nlayers=0)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

class Models.erfh5_ConvModel.S80Deconv2ToDrySpotEff(pretrained='', checkpoint_path=None, freeze_nlayers=0, round_at: float = None, demo_mode=False)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

class Models.erfh5_ConvModel.S80DeconvToDrySpotEff(pretrained='', checkpoint_path=None, freeze_nlayers=0)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

class Models.erfh5_ConvModel.SensorDeconvToDryspot(input_dim=1140)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

class Models.erfh5_ConvModel.SensorDeconvToDryspot2(pretrained=False, checkpoint_path=None, freeze_nlayers=0)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

load_model(path)

class Models.erfh5_ConvModel.SensorDeconvToDryspotEfficient(pretrained='', checkpoint_path=None, freeze_nlayers=0)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

class Models.erfh5_ConvModel.SensorDeconvToDryspotEfficient2(pretrained='', checkpoint_path=None, freeze_nlayers=0)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

class Models.erfh5_ConvModel.SensorToDryspotBoolModel

Bases: sphinx.ext.autodoc.importer._MockObject

forward(x)

class Models.erfh5_ConvModel.erfh5_Conv25D_Frame(sequence_len)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(x)

class Models.erfh5_ConvModel.erfh5_Conv2dPercentage

Bases: sphinx.ext.autodoc.importer._MockObject

forward(x)

class Models.erfh5_ConvModel.erfh5_Conv3d(sequence_len)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(x)

Models.erfh5_DeconvModel

class Models.erfh5_DeconvModel.DeconvModel(input_dim=1140)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

class Models.erfh5_DeconvModel.DeconvModel2x(input_dim=1140)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

class Models.erfh5_DeconvModel.DeconvModel4x(input_dim=1140)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

class Models.erfh5_DeconvModel.DeconvModel8x

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

class Models.erfh5_DeconvModel.DeconvModelEfficient

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

class Models.erfh5_DeconvModel.DeconvModelEfficientBn

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

class Models.erfh5_DeconvModel.S20DeconvModelEfficient(demo_mode=False)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

class Models.erfh5_DeconvModel.S80DeconvModelEfficient

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

class Models.erfh5_DeconvModel.S80DeconvModelEfficient2(pretrained='', checkpoint_path=None, freeze_nlayers=0, round_at: float = None, demo_mode=False)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

Models.erfh5_RNN_models

class Models.erfh5_RNN_models.ERFH5_RNN(input_dim, hidden_dim, batch_size, num_layers=1)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(x)

for i in range(len(hidden)): hidden[i] = hidden[i].permute(1, 0, 2).contiguous()

init_hidden()

init_weights()

class Models.erfh5_RNN_models.Net(input_dim, hidden_dim, batch_size, num_layers=4, encoder_path='/home', split_gpus=True, parallel=True)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(x, batch_size)

Models.erfh5_autoencoder

class Models.erfh5_autoencoder.StackedFullyConnected(FC_List=[500, 200, 100])

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

class Models.erfh5_autoencoder.erfh5_Autoencoder(input_size, FC_List=[500, 200, 100])

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inputs)

get_encoding()

class Models.erfh5_autoencoder.erfh5_Distributed_Autoencoder(dgx_mode=True, layers_size_list=[69366, 15000])

Bases: sphinx.ext.autodoc.importer._MockObject

forward(x)

save_encoder(path)

Models.erfh5_autoencoder.load_stacked_fc(path, list=[69366, 15000, 8192])

Models.erfh5_fullyConnected

class Models.erfh5_fullyConnected.S1140DryspotModelFCWide(input_dim=1140)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(_input)

class Models.erfh5_fullyConnected.S20DSModelFC4ChannelsFlatten(input_dim=80)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(_input)

class Models.erfh5_fullyConnected.S20DryspotModelFCWide(input_dim=20, demo_mode=False)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(_input)

class Models.erfh5_fullyConnected.S80DryspotModelFCWide(input_dim=80, demo_mode=False)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(_input)

class Models.erfh5_fullyConnected.SensorDryspotModelFC(input_dim=1140)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(inpt)

Models.erfh5_pressuresequence_CRNN

class Models.erfh5_pressuresequence_CRNN.ERFH5_PressureSequence_Model

Bases: sphinx.ext.autodoc.importer._MockObject

forward(x)

class Models.erfh5_pressuresequence_CRNN.ERFH5_Pressure_CNN(input_channels=50)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(x)

class Models.erfh5_pressuresequence_CRNN.ERFH5_Pressure_FC(input_features=24576)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(x)

init_weights()

class Models.erfh5_pressuresequence_CRNN.ERFH5_RNN(input_dim, hidden_dim=1024, batch_size=8, num_layers=4)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(x)

init_hidden()

init_weights()

Model.flow_front_to_fiber_fraction_model

class Models.flow_front_to_fiber_fraction_model.FlowfrontCNN

Bases: sphinx.ext.autodoc.importer._MockObject

forward(x)

class Models.flow_front_to_fiber_fraction_model.FlowfrontFeaturesRNN(input_dim, hidden_dim=512, batch_size=16, num_layers=3)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(x)

for i in range(len(hidden)): hidden[i] = hidden[i].permute(1, 0, 2).contiguous()

init_weights()

class Models.flow_front_to_fiber_fraction_model.FlowfrontToFiberfractionModel

Bases: sphinx.ext.autodoc.importer._MockObject

forward(x)

Models.custom_loss

File for special losses that are not implemented in PyTorch

class Models.custom_loss.FocalLoss(gamma=1, reduction='sum', pt_epsilon=0.0001)

Bases: sphinx.ext.autodoc.importer._MockObject

forward(p, y)

Models.custom_loss.focal_loss(p, y)

Models.model_utils

Models.model_utils.load_model_layers_from_path(path: pathlib.Path, layer_names: set)