simulation.utils.machine_learning.cycle_gan.models.cycle_gan_stats module

Summary

Classes:

CycleGANStats

Reference

class CycleGANStats(real_a: torch.Tensor = None, real_b: torch.Tensor = None, fake_a: torch.Tensor = None, fake_b: torch.Tensor = None, rec_a: torch.Tensor = None, rec_b: torch.Tensor = None, idt_a: torch.Tensor = None, idt_b: torch.Tensor = None, loss_g_a_to_b: float = None, loss_g_b_to_a: float = None, loss_idt_a: float = None, loss_idt_b: float = None, loss_cycle_a: float = None, loss_cycle_b: float = None, loss_d_a: float = None, loss_d_b: float = None, w_distance_a: float = None, w_distance_b: float = None)

Bases: object

real_a: torch.Tensor = None

real_b: torch.Tensor = None

fake_a: torch.Tensor = None

fake_b: torch.Tensor = None

rec_a: torch.Tensor = None

rec_b: torch.Tensor = None

idt_a: torch.Tensor = None

idt_b: torch.Tensor = None

loss_g_a_to_b: float = None

loss_g_b_to_a: float = None

loss_idt_a: float = None

loss_idt_b: float = None

loss_cycle_a: float = None

loss_cycle_b: float = None

loss_d_a: float = None

loss_d_b: float = None

w_distance_a: float = None

w_distance_b: float = None

get_visuals()

get_losses()