main.py

import torch  
import utility
import data
import model
import loss
from option import args
from trainer import Trainer

# To make sure that each time the-generade-random-number is fixed
torch.manual_seed(args.seed)

#print ("1-1-checkpoint")

# Open the module utility
checkpoint = utility.checkpoint(args)
#print ("1-2-checkpoint")

# 
if checkpoint.ok:
    print ("1-3-checkpoint")
    loader = data.Data(args,None)
    #print ("1-4-checkpoint")
    model = model.Model(args, checkpoint)
    #print ("1-5-checkpoint")
    loss = loss.Loss(args, checkpoint) if not args.test_only else None
    #print ("1-6-checkpoint")
    t = Trainer(args, loader, model, loss, checkpoint)
    #print ("1-7-checkpoint")
    while not t.terminate():
        t.train()
        t.test()
        #print ("1-8a-checkpoint")
    checkpoint.done()
    print ("1-8-checkpoint")
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 人面猴
    序言：七十年代末，一起剝皮案震驚了整個(gè)濱河市，隨后出現(xiàn)的幾起案子蕾额，更是在濱河造成了極大的恐慌，老刑警劉巖属划，帶你破解...
    沈念sama閱讀 211,123評(píng)論 6贊 490
  • 死咒
    序言：濱河連續(xù)發(fā)生了三起死亡事件，死亡現(xiàn)場(chǎng)離奇詭異候生，居然都是意外死亡同眯，警方通過(guò)查閱死者的電腦和手機(jī)，發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,031評(píng)論 2贊 384
  • 救了他兩次的神仙讓他今天三更去死
    文/潘曉璐 我一進(jìn)店門唯鸭，熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)须蜗，“玉大人，你說(shuō)我怎么就攤上這事∶靼梗” “怎么了菱农？”我有些...
    開(kāi)封第一講書(shū)人閱讀 156,723評(píng)論 0贊 345
  • 道士緝兇錄：失蹤的賣姜人
    文/不壞的土叔 我叫張陵，是天一觀的道長(zhǎng)柿估。 經(jīng)常有香客問(wèn)我循未，道長(zhǎng)，這世上最難降的妖魔是什么秫舌？ 我笑而不...
    開(kāi)封第一講書(shū)人閱讀 56,357評(píng)論 1贊 283
  • ?港島之戀（遺憾婚禮）
    正文 為了忘掉前任只厘，我火速辦了婚禮，結(jié)果婚禮上舅巷，老公的妹妹穿的比我還像新娘。我一直安慰自己河咽，他們只是感情好钠右，可當(dāng)我...
    茶點(diǎn)故事閱讀 65,412評(píng)論 5贊 384
  • 惡毒庶女頂嫁案：這布局不是一般人想出來(lái)的
    文/花漫 我一把揭開(kāi)白布。 她就那樣靜靜地躺著忘蟹，像睡著了一般飒房。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上媚值，一...
    開(kāi)封第一講書(shū)人閱讀 49,760評(píng)論 1贊 289
  • 城市分裂傳說(shuō)
    那天狠毯，我揣著相機(jī)與錄音，去河邊找鬼褥芒。 笑死嚼松，一個(gè)胖子當(dāng)著我的面吹牛，可吹牛的內(nèi)容都是我干的锰扶。 我是一名探鬼主播献酗，決...
    沈念sama閱讀 38,904評(píng)論 3贊 405
  • 雙鴛鴦連環(huán)套：你想象不到人心有多黑
    文/蒼蘭香墨 我猛地睜開(kāi)眼，長(zhǎng)吁一口氣：“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼坷牛！你這毒婦竟也來(lái)了罕偎？” 一聲冷哼從身側(cè)響起，我...
    開(kāi)封第一講書(shū)人閱讀 37,672評(píng)論 0贊 266
  • 萬(wàn)榮殺人案實(shí)錄
    序言：老撾萬(wàn)榮一對(duì)情侶失蹤京闰，失蹤者是張志新（化名）和其女友劉穎颜及，沒(méi)想到半個(gè)月后，有當(dāng)?shù)厝嗽跇?shù)林里發(fā)現(xiàn)了一具尸體蹂楣，經(jīng)...
    沈念sama閱讀 44,118評(píng)論 1贊 303
  • ?護(hù)林員之死
    正文 獨(dú)居荒郊野嶺守林人離奇死亡俏站，尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 36,456評(píng)論 2贊 325
  • ?白月光啟示錄
    正文 我和宋清朗相戀三年，在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了捐迫。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片乾翔。...
    茶點(diǎn)故事閱讀 38,599評(píng)論 1贊 340
  • 活死人
    序言：一個(gè)原本活蹦亂跳的男人離奇死亡，死狀恐怖，靈堂內(nèi)的尸體忽然破棺而出反浓，到底是詐尸還是另有隱情萌丈，我是刑警寧澤，帶...
    沈念sama閱讀 34,264評(píng)論 4贊 328
  • ?日本核電站爆炸內(nèi)幕
    正文 年R本政府宣布雷则，位于F島的核電站辆雾，受9級(jí)特大地震影響，放射性物質(zhì)發(fā)生泄漏月劈。R本人自食惡果不足惜度迂，卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,857評(píng)論 3贊 312
  • 男人毒藥：我在死后第九天來(lái)索命
    文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望猜揪。 院中可真熱鬧惭墓，春花似錦、人聲如沸而姐。這莊子的主人今日做“春日...
    開(kāi)封第一講書(shū)人閱讀 30,731評(píng)論 0贊 21
  • 一樁弒父案，背后竟有這般陰謀
    文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)拴念。三九已至钧萍，卻和暖如春，著一層夾襖步出監(jiān)牢的瞬間政鼠，已是汗流浹背风瘦。 一陣腳步聲響...
    開(kāi)封第一講書(shū)人閱讀 31,956評(píng)論 1贊 264
  • 情欲美人皮
    我被黑心中介騙來(lái)泰國(guó)打工， 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留公般，地道東北人万搔。 一個(gè)月前我還...
    沈念sama閱讀 46,286評(píng)論 2贊 360
  • 代替公主和親
    正文 我出身青樓，卻偏偏與公主長(zhǎng)得像官帘，于是被迫代替她去往敵國(guó)和親蟹略。 傳聞我的和親對(duì)象是個(gè)殘疾皇子，可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 43,465評(píng)論 2贊 348

推薦閱讀更多精彩內(nèi)容