def __init__(self, args): self.tpu = False if args.device == 'cpu': self.device = 'cpu' elif args.device == 'gpu': self.device = 'cuda:0' elif args.device == 'tpu': self.device = xm.xla_device() self.tpu = True else: exit(0) #torch.manual_seed(777) self.save_dir = args.save_dir self.start_epoch = 0 self.summary_write = args.summary_write self.save_model = args.save_model self.end_epoch = args.end_epoch folder([ self.save_dir, self.save_dir + '/models', self.save_dir + '/runs' ]) self.summary = SummaryWriter( self.save_dir + '/runs/' + datetime.now(timezone).strftime("%Y-%m-%d-%H%M")) self.dataset_train = DiabetesDataset_CNN(train=True) self.dataset_test = DiabetesDataset_CNN(train=False) self.train_loader = DataLoader(dataset=self.dataset_train, batch_size=32, shuffle=True, num_workers=0) self.evaluate_loader = DataLoader(dataset=self.dataset_test, batch_size=32, shuffle=False, num_workers=0) self.model = CNN() self.model = self.model.to(self.device) if args.resume is not None: self.load_checkpoint() self.criterion = nn.CrossEntropyLoss().to(self.device) self.optimizer = torch.optim.Adam(self.model.parameters(), lr=0.001)
def __init__(self, fileName, addon): folderPath = utils.folder() self.addon = addon self.filePath = os.path.join(self.basePath, fileName) if fileName == utils.get_setting(addon_id,'xmltv.url'): self.fileUrl = fileName self.fileName = fileName.split('/')[-1] self.filePath = os.path.join(self.basePath, 'custom.xml') elif fileName == utils.get_setting(addon_id,'xmltv.url') and fileName.endswith(".zip"): self.fileUrl = fileName self.fileName = fileName.split('/')[-1] self.filePath = os.path.join(self.basePath, 'custom.xml') elif fileName == utils.get_setting(addon_id,'sub.xmltv.url'): self.fileUrl = fileName self.fileName = fileName.split('/')[-1] self.filePath = os.path.join(self.basePath, utils.get_setting(addon_id,'sub.xmltv')+'.xml') else: self.fileUrl = folderPath + fileName.replace('.xml', '.zip') # make sure the folder is actually there already! if not os.path.exists(self.basePath): os.makedirs(self.basePath)
def main(args): matrices_path = glob.glob(utils.folder(args.matrices_folder) + "*.txt") output_file = args.output_file ids_path = None if args.ids_path: ids_path = utils.folder(args.ids_path) c = Corpus(matrices_path, args.target, ids_path) if args.output_folder: output_folder = utils.folder(args.output_folder) c.write_invidual_files(output_folder) #c.write_summary(output_file + ".summary") #c.write_buckets(output_file + ".classes") #c.write_buckets(output_file + ".classes.verbose", verbose=True) if ids_path: c.write_zrc_buckets(utils.folder(args.segmentation_path), utils.folder(args.labs_path), output_file, BUCKETS)
import shutil import utils import xbmcvfs addon_id = 'script.ivueguide' addon_name = xbmcaddon.Addon(addon_id) HOME = addon_name.getAddonInfo('path') ICON = os.path.join(HOME, 'icon.png') PACKAGES = xbmc.translatePath( os.path.join('special://home', 'addons', 'packages')) TEMP = addon_name.getSetting('tempskin') USER_AGENT = 'Mozilla/5.0 (Windows; U; Windows NT 5.1; en-GB; rv:1.9.0.3) Gecko/2008092417 Firefox/3.0.3' #Karls changes ivue = utils.folder() d = xbmcgui.Dialog() dp = xbmcgui.DialogProgress() skin = addon_name.getSetting('skin') path = xbmc.translatePath( os.path.join('special://profile', 'addon_data', 'script.ivueguide', 'resources', 'skins')) Skinxml = xbmc.translatePath( os.path.join(path, skin, '720p', 'script-tvguide-main.xml')) xmlData = xbmc.translatePath( 'special://profile/addon_data/script.ivueguide/resources/config/Data.txt') logoData = xbmc.translatePath( 'special://profile/addon_data/script.ivueguide/resources/config/Logo.txt') catData = xbmc.translatePath( 'special://profile/addon_data/script.ivueguide/resources/categories/') ivue = utils.folder()