plots_file = os.path.join('plots', save_str) # relevant variables num_workers = 4 N0 = 2048 N1 = 1000 N2 = 500 num_classes = len(classes) init_lr = 0.0001 num_epochs = 3 num_clusters = kmeans.n_clusters kmeans_dict = kmeans.cluster_centers_ # DATA # datasets real_data = MultibinImages(augmented_path, 'real', 'c0', kmeans_file) render_data = MultibinImages(render_path, 'render', 'c0', kmeans_file) test_data = Pascal3dAll(pascal3d_path, 'test') # setup data loaders real_loader = DataLoader(real_data, batch_size=4, shuffle=True, num_workers=num_workers, pin_memory=True, collate_fn=my_collate) render_loader = DataLoader(render_data, batch_size=4, shuffle=True, num_workers=num_workers, pin_memory=True, collate_fn=my_collate)
ndim = 3 num_workers = 4 N0 = 2048 N1 = 1000 N2 = 500 N3 = 100 num_classes = len(classes) init_lr = 0.0001 num_epochs = 3 problem_type = 'm0' criterion = loss_m0(1.0) # DATA # datasets real_data = MultibinImages(augmented_path, 'real', problem_type, kmeans_file) render_data = MultibinImages(render_path, 'render', problem_type, kmeans_file) test_data = Pascal3dAll(pascal3d_path, 'test') # setup data loaders real_loader = DataLoader(real_data, batch_size=4, shuffle=True, num_workers=num_workers, pin_memory=True, collate_fn=my_collate) render_loader = DataLoader(render_data, batch_size=4, shuffle=True, num_workers=num_workers, pin_memory=True, collate_fn=my_collate)
log_dir = os.path.join('logs', save_str) # relevant variables num_workers = 4 N0 = 2048 N1 = 1000 N2 = 500 num_classes = len(classes) init_lr = 0.0001 num_epochs = 3 num_clusters = kmeans.n_clusters kmeans_dict = kmeans.cluster_centers_ # DATA # datasets real_data = MultibinImages(pascal3d_path, 'real', 'c0', kmeans_file) render_data = MultibinImages(pascal3d_path, 'render', 'c0', kmeans_file) test_data = Pascal3dAll(pascal3d_path, 'test') # setup data loaders real_loader = DataLoader(real_data, batch_size=4, shuffle=True, num_workers=num_workers, pin_memory=True, collate_fn=my_collate) render_loader = DataLoader(render_data, batch_size=4, shuffle=True, num_workers=num_workers, pin_memory=True, collate_fn=my_collate)