ETRI IVCL 1.0.0
Acceleration SW Platform for Ondevice
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
Namespaces | Functions | Variables
train.py File Reference

Namespaces

namespace  train
 

Functions

def train.save_checkpoint (save_path, epoch, depthNet, optimizer, is_best, filename='checkpoint.pth')
 model save method #### More...
 
def train.save_urgentstop (save_path, depthNet_state, filename='urgentstop.pth')
 
def train.save_path_formatter (args, parser)
 
def train.main ()
 
def train.np2Depth (input_tensor, invaild_mask)
 
def train.showdepth (predict_depths, gt_depth, b_idx)
 
def train.train (epoch, train_loader, HCDepthNet, optimizer, epoch_length, train_writer)
 
def train.validate (epoch, val_loader, HCDepthNet, epoch_length, validating_writer)
 

Variables

 train.parser
 
 train.metavar
 
 train.help
 
 train.type
 
 train.int
 
 train.default
 
 train.float
 
 train.dest
 
 train.None
 
 train.action
 
bool train.scaling = True
 
int train.best_error = -1
 
 train.device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 
int train.scaled_width = 480
 
int train.scaled_height = 352
 
float train.dmin = 0.5
 
float train.dmax = 32.0
 
int train.level = 64